ai-parrot 0.5.13__tar.gz → 0.5.15__tar.gz

{ai_parrot-0.5.13 → ai_parrot-0.5.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ai-parrot
-Version: 0.5.13
+Version: 0.5.15
 Summary: Live Chatbots based on Langchain chatbots and Agents     Integrated into Navigator Framework or used into any aiohttp applications.
 Home-page: https://github.com/phenobarbital/ai-parrot
 Author: Jesus Lara

{ai_parrot-0.5.13 → ai_parrot-0.5.15}/parrot/bots/abstract.py

@@ -540,11 +540,12 @@ class AbstractBot(DBInterface, ABC):
                 f"Error clearing chat history: {e}"
             )
 
-    def get_response(self, response: dict):
+    def get_response(self, response: dict, query: str = None):
         if 'error' in response:
             return response  # return this error directly
         try:
             response = ChatResponse(**response)
+            response.query = query
             response.response = self.as_markdown(
                 response,
                 return_sources=self.return_sources
@@ -697,15 +698,15 @@ class AbstractBot(DBInterface, ABC):
         )
         # re-configure LLM:
         new_llm = kwargs.pop('llm', None)
-        llm_config = kwargs.pop(
-            'llm_config',
-            {
-                "temperature": 0.2,
-                "top_k": 30,
-                "Top_p": 0.6
-            }
-        )
         if new_llm:
+            llm_config = kwargs.pop(
+                'llm_config',
+                {
+                    "temperature": 0.2,
+                    "top_k": 30,
+                    "Top_p": 0.6
+                }
+            )
             self.configure_llm(llm=new_llm, config=llm_config)
         # Combine into a ChatPromptTemplate
         prompt = PromptTemplate(
@@ -765,11 +766,11 @@ class AbstractBot(DBInterface, ABC):
                 "query": question,
                 "error": str(e)
             }
-        return self.get_response(response)
+        return self.get_response(response, question)
 
     def as_markdown(self, response: ChatResponse, return_sources: bool = False) -> str:
         markdown_output = f"**Question**: {response.question}  \n"
-        markdown_output += f"**Answer**: {response.answer}  \n"
+        markdown_output += f"**Answer**: \n {response.answer}  \n"
         if return_sources is True and response.source_documents:
             source_documents = response.source_documents
             current_sources = []
@@ -968,23 +969,25 @@ class AbstractBot(DBInterface, ABC):
             )
         async with self.store as store:  #pylint: disable=E1101
             print('STORES > ', self.stores, store)
+            # Check if we have multiple stores:
             if self._use_vector:
                 if len(self.stores) > 1:
-                    retriever = MultiVectorStoreRetriever(
-                        stores=self.stores,
-                        metric_type=metric_type,
-                        search_type=search_type,
-                        chain_type=chain_type,
-                        search_kwargs=search_kwargs
-                    )
-                else:
-                    vector = store.get_vector(metric_type=metric_type)
-                    retriever = VectorStoreRetriever(
-                        vectorstore=vector,
-                        search_type=search_type,
-                        chain_type=chain_type,
-                        search_kwargs=search_kwargs
-                    )
+                    store = self.stores[0]
+                #     retriever = MultiVectorStoreRetriever(
+                #         stores=self.stores,
+                #         metric_type=metric_type,
+                #         search_type=search_type,
+                #         chain_type=chain_type,
+                #         search_kwargs=search_kwargs
+                #     )
+                # else:
+                vector = store.get_vector(metric_type=metric_type)
+                retriever = VectorStoreRetriever(
+                    vectorstore=vector,
+                    search_type=search_type,
+                    chain_type=chain_type,
+                    search_kwargs=search_kwargs
+                )
             else:
                 retriever = EmptyRetriever()
             if self.kb:
@@ -1011,7 +1014,7 @@ class AbstractBot(DBInterface, ABC):
                 response = await chain.ainvoke(
                     {"question": question}
                 )
-                return self.get_response(response)
+                return self.get_response(response, question)
             except asyncio.CancelledError:
                 # Handle task cancellation
                 print("Conversation task was cancelled.")

{ai_parrot-0.5.13 → ai_parrot-0.5.15}/parrot/version.py

@@ -3,7 +3,7 @@
 __title__ = "ai-parrot"
 __description__ = "Live Chatbots based on Langchain chatbots and Agents \
     Integrated into Navigator Framework or used into any aiohttp applications."
-__version__ = "0.5.13"
+__version__ = "0.5.15"
 __author__ = "Jesus Lara"
 __author_email__ = "jesuslarag@gmail.com"
 __license__ = "MIT"