adam/meshing-around
1
0
Fork 0
mirror of https://github.com/SpudGunMan/meshing-around.git synced 2026-08-06 17:03:04 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update llm.py

Browse Source
This commit is contained in:
SpudGunMan
2024-10-10 00:11:01 -07:00
parent f5512b19da
commit a81fafe268
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/llm.py
+4 -2
View File
@@ -144,8 +144,10 @@ def llm_query(input, nodeID=0, location_name=None):
# #ragQuery = langchain.generate_prompt(modelPrompt)
# Query the model with RAG context
#result = ollamaClient.generate(model=llmModel, prompt=modelPrompt, context=ragContext)
result = ollamaClient.generate(model=llmModel, prompt=modelPrompt)
if ragContext:
result = ollamaClient.generate(model=llmModel, prompt=modelPrompt, context=ragContext)
else:
result = ollamaClient.generate(model=llmModel, prompt=modelPrompt)
else:
# Query the model without RAG context
result = ollamaClient.generate(model=llmModel, prompt=modelPrompt)