Update llm.py

This commit is contained in:
SpudGunMan
2024-10-10 00:00:33 -07:00
parent 6da457bb06
commit f5512b19da
+2 -1
View File
@@ -144,7 +144,8 @@ 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, context=ragContext)
result = ollamaClient.generate(model=llmModel, prompt=modelPrompt)
else:
# Query the model without RAG context
result = ollamaClient.generate(model=llmModel, prompt=modelPrompt)