mb-rag 1.1.13__py3-none-any.whl → 1.1.18__py3-none-any.whl

mb_rag/chatbot/basic.py

@@ -145,6 +145,7 @@ class ModelFactory:
             raise ImportError("Langchain Community package not found. Please install it using: pip install langchain_ollama")
         
         from langchain_ollama import ChatOllama
+        print(f"Current Ollama serve model is {os.system('ollama ps')}")
         kwargs["model"] = model_name
         return ChatOllama(**kwargs)
 
@@ -460,3 +461,105 @@ class IPythonStreamHandler(StreamingStdOutCallbackHandler):
         """Handle new token"""
         self.output += token
         display(HTML(self.output), clear=True)
+
+
+class AgentFactory:
+    """Factory class for creating different types of agents"""
+
+    def __init__(self, agent_type: str = 'basic', model_name: str = "gpt-4o", **kwargs) -> Any:
+        """
+        Factory method to create any type of agent
+        Args:
+            agent_type (str): Type of agent to create. Default is basic.
+            model_name (str): Name of the model
+            **kwargs: Additional arguments
+        Returns:
+            Any: Agent
+        """
+        creators = {
+            'basic': self.create_basic_agent,
+            'langgraph': self.create_langgraph_agent,
+        }
+
+        agent_data = creators.get(agent_type)
+        if not agent_data:
+            raise ValueError(f"Unsupported agent type: {agent_type}")
+
+        try:
+            self.agent = agent_data(model_name, **kwargs)
+        except Exception as e:
+            raise ValueError(f"Error creating {agent_type} agent: {str(e)}")
+
+    @classmethod
+    def create_basic_agent(cls, model_name: str = "gpt-4o", **kwargs) -> Any:
+        """
+        Create basic agent
+        Args:
+            model_name (str): Name of the model
+            **kwargs: Additional arguments
+        Returns:
+            Runnable: Agent
+        """
+        # Basic agent creation logic here
+        llm = ModelFactory(model_name=model_name, **kwargs).model
+        from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
+        prompt = ChatPromptTemplate.from_messages([
+            ("system", "You are a helpful AI assistant"),
+            MessagesPlaceholder(variable_name="messages")
+        ])
+        from langchain_core.runnables import chain
+        agent = prompt | llm
+        return agent
+
+    @classmethod
+    def create_langgraph_agent(cls, model_name: str = "gpt-4o", **kwargs) -> Any:
+        """
+        Create LangGraph agent
+        Args:
+            model_name (str): Name of the model
+            **kwargs: Additional arguments
+        Returns:
+            Graph: LangGraph agent
+        """
+        if not check_package("langgraph"):
+            raise ImportError("LangGraph package not found. Please install it using: pip install langgraph")
+
+        from langgraph.graph import StateGraph, MessageGraph
+        from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
+        from langchain_core.runnables import chain
+        from langchain_core.messages import BaseMessage
+
+        llm = ModelFactory(model_name=model_name, **kwargs).model
+
+        # Define the state of the graph
+        class GraphState:
+            messages: List[BaseMessage]
+            agent_state: Dict[str, Any]
+
+        # Define the nodes
+        def agent(state: GraphState):
+            prompt = ChatPromptTemplate.from_messages([
+                ("system", "You are a helpful AI assistant"),
+                MessagesPlaceholder(variable_name="messages")
+            ])
+            return (prompt | llm).invoke({"messages": state.messages})
+
+        def user(state: GraphState, input: str):
+            return HumanMessage(content=input)
+
+        # Define the graph
+        graph = MessageGraph()
+
+        # Add the nodes
+        graph.add_node("agent", agent)
+        graph.add_node("user", user)
+
+        # Set the entrypoint
+        graph.set_entry_point("user")
+
+        # Add the edges
+        graph.add_edge("user", "agent")
+        graph.add_edge("agent", "user")
+
+        # Compile the graph
+        return graph.compile()

mb_rag/version.py

@@ -1,5 +1,5 @@
 MAJOR_VERSION = 1
 MINOR_VERSION = 1
-PATCH_VERSION = 13
+PATCH_VERSION = 18
 version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
 __all__  = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']

{mb_rag-1.1.13.dist-info → mb_rag-1.1.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mb_rag
-Version: 1.1.13
+Version: 1.1.18
 Summary: RAG function file
 Author: ['Malav Bateriwala']
 Requires-Python: >=3.8

{mb_rag-1.1.13.dist-info → mb_rag-1.1.18.dist-info}/RECORD

@@ -1,7 +1,7 @@
 mb_rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/version.py,sha256=ADdU2A-QIuJ5Dlo6U75iAj6ZRCbWMl8r62U0V7373Hc,207
+mb_rag/version.py,sha256=kHqa0xhMfGNm0aT1_krDNIuXwX8m3fsXu737vjLlKwA,207
 mb_rag/chatbot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/chatbot/basic.py,sha256=avzYnqwmEqTfwms98ocomjNgJxJ9lv-xGbZO55PlJx8,16873
+mb_rag/chatbot/basic.py,sha256=8uHwTZ8YepGtphTE0pd2T8p51xzthmtlFmzfIpiK6A0,20395
 mb_rag/chatbot/chains.py,sha256=vDbLX5R29sWN1pcFqJ5fyxJEgMCM81JAikunAEvMC9A,7223
 mb_rag/chatbot/prompts.py,sha256=n1PyiLbU-5fkslRv6aVOzt0dDlwya_cEdQ7kRnRhMuY,1749
 mb_rag/rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -9,7 +9,7 @@ mb_rag/rag/embeddings.py,sha256=KjBdekFDb5M3dRMco4r3dDMXMsG5dxdzKImuVIipsd0,2709
 mb_rag/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mb_rag/utils/bounding_box.py,sha256=G0hdDam8QmYtD9lfwMeDHGm-TTo6KZg-yK5ESFL9zaM,8366
 mb_rag/utils/extra.py,sha256=spbFrGgdruNyYQ5PzgvpSIa6Nm0rn9bb4qc8W9g582o,2492
-mb_rag-1.1.13.dist-info/METADATA,sha256=5Evo710iPFcejrhkVzAFaJL3RiNqXluzjC5-nqD40j8,234
-mb_rag-1.1.13.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-mb_rag-1.1.13.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
-mb_rag-1.1.13.dist-info/RECORD,,
+mb_rag-1.1.18.dist-info/METADATA,sha256=YM6qxjtilsWx2eTouiQTo-8uJavCjxe0IShAwJ4bOBg,234
+mb_rag-1.1.18.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+mb_rag-1.1.18.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
+mb_rag-1.1.18.dist-info/RECORD,,