diff --git a/modules/llm.py b/modules/llm.py index 3ed1960..c47fd5d 100644 --- a/modules/llm.py +++ b/modules/llm.py @@ -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)