From f5512b19da98183395758c405b6d700d21ce31a0 Mon Sep 17 00:00:00 2001 From: SpudGunMan Date: Thu, 10 Oct 2024 00:00:33 -0700 Subject: [PATCH] Update llm.py --- modules/llm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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)