ai-parrot 0.2.4__tar.gz → 0.2.6__tar.gz

{ai_parrot-0.2.4 → ai_parrot-0.2.6}/PKG-INFO

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

{ai_parrot-0.2.4 → ai_parrot-0.2.6}/ai_parrot.egg-info/PKG-INFO

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

{ai_parrot-0.2.4 → ai_parrot-0.2.6}/parrot/chatbots/retrievals/__init__.py

@@ -84,26 +84,27 @@ class RetrievalManager:
         # Web Request:
         self.request = request
 
-
     def __enter__(self):
-        self.client_id: str = str(uuid.uuid4())
-        self.client, self.client_id = self.store.connect()
+        # self.client_id: str = str(uuid.uuid4())
+        # self.client, self.client_id = self.store.connect()
         return self
 
     async def __aenter__(self):
-        self.client_id: str = str(uuid.uuid4())
-        self.client, self.client_id = self.store.connect()
+        # self.client_id: str = str(uuid.uuid4())
+        # self.client, self.client_id = self.store.connect()
         return self
 
     async def __aexit__(self, exc_type, exc_value, traceback):
-        self.client.close()  # closing database connection
+        #self.client.close()  # closing database connection
         # closing the connection:
-        self.store.close(client_id=self.client_id)
+        # self.store.close(alias=self.client_id)
+        pass
 
     def __exit__(self, exc_type, exc_value, traceback):
-        self.client.close()  # closing database connection
+        # self.client.close()  # closing database connection
         # closing the connection:
-        self.store.close(client_id=self.client_id)
+        # self.store.close(alias=self.client_id)
+        pass
 
     def create_memory(
         self,

{ai_parrot-0.2.4 → ai_parrot-0.2.6}/parrot/handlers/chat.py

@@ -98,13 +98,6 @@ class ChatHandler(BaseView):
                     question=question,
                     memory=memory
                 )
-                # qa = retrieval.qa(
-                #     question=question,
-                #     search_kwargs={"k": 10},
-                #     use_llm=llm,
-                #     # memory=memory
-                # )
-                # result = await qa.invoke(question)
                 # Drop "memory" information:
                 result.chat_history = None
                 result.source_documents = None

{ai_parrot-0.2.4 → ai_parrot-0.2.6}/parrot/stores/abstract.py

@@ -47,6 +47,7 @@ class AbstractStore(ABC):
         self.logger = logging.getLogger(f"Store.{__name__}")
         # client
         self._client = None
+        self._client_id = None
 
     @property
     def connected(self) -> bool:
@@ -61,7 +62,7 @@ class AbstractStore(ABC):
             self._embed_ = self.create_embedding(
                 model_name=self.embedding_name
             )
-        self._client, self.client_id = self.connect()
+        self._client, self._client_id = self.connect()
         return self
 
     @abstractmethod

{ai_parrot-0.2.4 → ai_parrot-0.2.6}/parrot/stores/milvus.py

@@ -163,19 +163,19 @@ class MilvusStore(AbstractStore):
         self._embed_ = None
         del self.tensor
         try:
-            self.close(alias=self.client_id)
+            self.close(alias=self._client_id)
             torch.cuda.empty_cache()
         except RuntimeError:
             pass
 
-    async def connection(self, alias: str = None):
+    def connection(self, alias: str = None):
         if not alias:
             # self._client_id = str(uuid.uuid4())
             self._client_id = 'uri-connection'
         else:
             self._client_id = alias
         # making the connection:
-        self._client, self.client_id = self.connect(
+        self._client, self._client_id = self.connect(
             alias=self._client_id
         )
         return self

{ai_parrot-0.2.4 → ai_parrot-0.2.6}/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 aiohttp applications."
-__version__ = "0.2.4"
+__version__ = "0.2.6"
 __author__ = "Jesus Lara"
 __author_email__ = "jesuslarag@gmail.com"
 __license__ = "MIT"