sonika-langchain-bot 0.0.8__py3-none-any.whl → 0.0.10__py3-none-any.whl

sonika_langchain_bot/langchain_bdi.py

@@ -64,16 +64,6 @@ class BotBeliefSystem:
     def _initialize_core_beliefs(self):
         """Initializes the core beliefs of the bot."""
         core_beliefs = [
-            Belief(
-                f"My name is {self.bot_name}.",
-                BeliefType.CORE,
-                confidence=1.0
-            ),
-            Belief(
-                f"I operate on Telegram, and my responses should be as concise and accurate as possible.",
-                BeliefType.CORE,
-                confidence=1.0
-            ),
             # Add more core beliefs as needed
         ]
         
@@ -81,6 +71,9 @@ class BotBeliefSystem:
             self.add_belief(belief)
 
     def add_tools_beliefs(self, tools: List[Tool]):
+
+        if len(tools) == 0:
+            return
         """Carga y procesa las herramientas disponibles para el bot."""
         # Instrucciones sobre el uso de herramientas en inglés
         instrucciones_de_uso = '''\nWhen you want to execute a tool, enclose the command with three asterisks and provide all parameters needed.

sonika_langchain_bot/langchain_bot_agent_bdi.py

@@ -1,3 +1,4 @@
+from typing import Generator
 from typing import List
 from langchain.memory import ConversationBufferMemory
 from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder, SystemMessagePromptTemplate, HumanMessagePromptTemplate
@@ -45,6 +46,7 @@ class LangChainBot:
         """
         beliefs_context = self.belief_system.generate_prompt_context()
         full_system_prompt = f"{beliefs_context}\n\n"
+        print(full_system_prompt)
 
         prompt = ChatPromptTemplate.from_messages([
             SystemMessagePromptTemplate.from_template(full_system_prompt),
@@ -93,6 +95,8 @@ class LangChainBot:
         if context:
             augmented_input = f"Context from attached files:\n{context}\n\nUser question: {user_input}"
 
+        print(augmented_input)
+
         bot_response = self.conversation.invoke({"input": augmented_input, "history": self.memory.chat_memory.messages})
 
         token_usage= bot_response.response_metadata['token_usage']
@@ -128,6 +132,26 @@ class LangChainBot:
         self.save_messages(user_input, bot_response)
 
         return ResponseModel(user_tokens=user_tokens, bot_tokens=bot_tokens, response=bot_response)
+    
+    def get_response_stream(self, user_input: str) -> Generator[str, None, None]:
+        """
+        Genera una respuesta en streaming para la entrada del usuario, procesando el contexto.
+
+        Args:
+            user_input (str): Entrada del usuario
+
+        Yields:
+            str: Fragmentos de la respuesta generada por el modelo en tiempo real
+        """
+        context = self._get_context(user_input)
+        augmented_input = f"User question: {user_input}"
+        if context:
+            augmented_input = f"Context from attached files:\n{context}\n\nUser question: {user_input}"
+
+        result_stream = self.conversation.stream({"input": augmented_input, "history": self.memory.chat_memory.messages})
+        for response in result_stream:
+            yield response.content
+
 
     def _get_context(self, query: str) -> str:
         """

sonika_langchain_bot/langchain_models.py

@@ -1,8 +1,7 @@
 from langchain_openai import ChatOpenAI
 from sonika_langchain_bot.langchain_class import ILanguageModel
 from langchain_core.messages import HumanMessage
-from typing import  Type
-from pydantic import BaseModel
+from typing import Generator
 
 
 class OpenAILanguageModel(ILanguageModel):
@@ -46,4 +45,18 @@ class OpenAILanguageModel(ILanguageModel):
         """
         message = HumanMessage(content=prompt)
         response = self.model.invoke([message])
-        return response
+        return response
+    
+    def stream_response(self, prompt: str) -> Generator[str, None, None]:
+        """
+        Genera una respuesta en streaming basada en el prompt proporcionado.
+
+        Args:
+            prompt (str): Texto de entrada para generar la respuesta
+        
+        Yields:
+            str: Fragmentos de la respuesta generada por el modelo en tiempo real
+        """
+        message = HumanMessage(content=prompt)
+        for chunk in self.model.stream([message]):
+            yield chunk.content

{sonika_langchain_bot-0.0.8.dist-info → sonika_langchain_bot-0.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sonika-langchain-bot
-Version: 0.0.8
+Version: 0.0.10
 Summary: Agente langchain con LLM
 Author: Erley Blanco Carvajal
 License: MIT License
@@ -30,7 +30,8 @@ Provides-Extra: dev
 Requires-Dist: sphinx <9.0.0,>=8.1.3 ; extra == 'dev'
 Requires-Dist: sphinx-rtd-theme <4.0.0,>=3.0.1 ; extra == 'dev'
 
-# Sonika LangChain Bot
+# Sonika LangChain Bot <a href="https://pepy.tech/projects/sonika-langchain-bot"><img src="https://static.pepy.tech/badge/sonika-langchain-bot" alt="PyPI Downloads"></a>
+
 
 Una librería Python que implementa un bot conversacional utilizando LangChain con capacidades BDI (Belief-Desire-Intention) y clasificación de texto.
 
@@ -104,6 +105,10 @@ bot_response = response_model
 
 print(bot_response)
 
+#o por streaming
+for chunk in bot.get_response_stream(user_message):
+    print(chunk)
+
 ```
 
 ### Ejemplo de Clasificación de Texto
@@ -137,3 +142,4 @@ result = classifier.classify("Tu texto aquí")
 ## Contribución
 
 Las contribuciones son bienvenidas. Por favor, abre un issue para discutir los cambios importantes que te gustaría hacer.
+

sonika_langchain_bot-0.0.10.dist-info/RECORD

@@ -0,0 +1,13 @@
+sonika_langchain_bot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sonika_langchain_bot/langchain_bdi.py,sha256=ithc55azP5XSPb8AGRUrDGYnVI6I4IqpqElLNat4BAQ,7024
+sonika_langchain_bot/langchain_bot_agent_bdi.py,sha256=Ev0hhRQYe6kyGAHiFDhFsfu6QnTwUFaA9oB8DfNV7u4,8613
+sonika_langchain_bot/langchain_clasificator.py,sha256=GR85ZAliymBSoDa5PXB31BvJkuiokGjS2v3RLdXnzzk,1381
+sonika_langchain_bot/langchain_class.py,sha256=5anB6v_wCzEoAJRb8fV9lPPS72E7-k51y_aeiip8RAw,1114
+sonika_langchain_bot/langchain_files.py,sha256=SEyqnJgBc_nbCIG31eypunBbO33T5AHFOhQZcghTks4,381
+sonika_langchain_bot/langchain_models.py,sha256=vqSSZ48tNofrTMLv1QugDdyey2MuIeSdlLSD37AnzkI,2235
+sonika_langchain_bot/langchain_tools.py,sha256=y7wLf1DbUua3QIvz938Ek-JIMOuQhrOIptJadW8OIsU,466
+sonika_langchain_bot-0.0.10.dist-info/LICENSE,sha256=O8VZ4aU_rUMAArvYTm2bshcZ991huv_tpfB5BKHH9Q8,1064
+sonika_langchain_bot-0.0.10.dist-info/METADATA,sha256=UFs95OjxiI17n839Ylb2ou-eR8sJ11MrMu7V3TOa0sA,4455
+sonika_langchain_bot-0.0.10.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+sonika_langchain_bot-0.0.10.dist-info/top_level.txt,sha256=UsTTSZFEw2wrPSVh4ufu01e2m_E7O_QVYT_k4zCQaAE,21
+sonika_langchain_bot-0.0.10.dist-info/RECORD,,

sonika_langchain_bot-0.0.8.dist-info/RECORD

@@ -1,13 +0,0 @@
-sonika_langchain_bot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sonika_langchain_bot/langchain_bdi.py,sha256=aCo6d2vhn5d1xGmN3Ldpa8DStJUSLuhDPZ3y4iXILIg,7329
-sonika_langchain_bot/langchain_bot_agent_bdi.py,sha256=4CjeIDOTefb7DzTq3W7Fq-jhSGpBgaR1WuDr_xNT5Ig,7721
-sonika_langchain_bot/langchain_clasificator.py,sha256=GR85ZAliymBSoDa5PXB31BvJkuiokGjS2v3RLdXnzzk,1381
-sonika_langchain_bot/langchain_class.py,sha256=5anB6v_wCzEoAJRb8fV9lPPS72E7-k51y_aeiip8RAw,1114
-sonika_langchain_bot/langchain_files.py,sha256=SEyqnJgBc_nbCIG31eypunBbO33T5AHFOhQZcghTks4,381
-sonika_langchain_bot/langchain_models.py,sha256=AglQNSFXH5cVudBmtSAQ4t51piRXmJfrCfDhGULLt0I,1761
-sonika_langchain_bot/langchain_tools.py,sha256=y7wLf1DbUua3QIvz938Ek-JIMOuQhrOIptJadW8OIsU,466
-sonika_langchain_bot-0.0.8.dist-info/LICENSE,sha256=O8VZ4aU_rUMAArvYTm2bshcZ991huv_tpfB5BKHH9Q8,1064
-sonika_langchain_bot-0.0.8.dist-info/METADATA,sha256=RfOmBFZ9t5E3GMxh8NnDj9SHCpeykyOgGgHuaeqWiXI,4218
-sonika_langchain_bot-0.0.8.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-sonika_langchain_bot-0.0.8.dist-info/top_level.txt,sha256=UsTTSZFEw2wrPSVh4ufu01e2m_E7O_QVYT_k4zCQaAE,21
-sonika_langchain_bot-0.0.8.dist-info/RECORD,,