diff --git a/src/shoppingassistantservice/shoppingassistantservice.py b/src/shoppingassistantservice/shoppingassistantservice.py index 49de86bcb18..809c49e2e09 100755 --- a/src/shoppingassistantservice/shoppingassistantservice.py +++ b/src/shoppingassistantservice/shoppingassistantservice.py @@ -67,7 +67,6 @@ def talkToGemini(): print("Beginning RAG call") prompt = request.json['message'] prompt = unquote(prompt) - # Step 1 – Get a room description from Gemini-vision-pro llm_vision = ChatGoogleGenerativeAI(model="gemini-1.5-flash") message = HumanMessage( @@ -93,6 +92,7 @@ def talkToGemini(): print(f"Retrieved documents: {len(docs)}") #Prepare relevant documents for inclusion in final prompt relevant_docs = "" + talkTogemini() for doc in docs: doc_details = doc.to_json() print(f"Adding relevant document to prompt context: {doc_details}")