From d03d7dbc470b1232dbab5b531b864375d93aca54 Mon Sep 17 00:00:00 2001 From: SpudGunMan Date: Mon, 21 Oct 2024 19:36:02 -0700 Subject: [PATCH] Update llm.py --- modules/llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/llm.py b/modules/llm.py index dd57ae3..138b554 100644 --- a/modules/llm.py +++ b/modules/llm.py @@ -190,7 +190,7 @@ def llm_query(input, nodeID=0, location_name=None): # Build the query from the template modelPrompt = meshBotAI.format(input=input, context=ragContext, location_name=location_name, llmModel=llmModel, history=history) # Query the model with RAG context - result = ollamaClient.generate(model=llmModel, prompt=f"Using this data: {ragContext}. Respond to this prompt: {input}") + result = ollamaClient.generate(model=llmModel, prompt=modelPrompt) else: # Build the query from the template modelPrompt = meshBotAI.format(input=input, context='\n'.join(googleResults), location_name=location_name, llmModel=llmModel, history=history)