add: github integration with mcp server

main.py

@@ -9,7 +9,7 @@ from langgraph.graph import MessagesState, END, StateGraph
 from langchain_core.messages import HumanMessage
 from langgraph.checkpoint.memory import MemorySaver
 
-from pmcp.agents.trello_agent import TrelloAgent
+from pmcp.agents.pm_agent import ProjectManagerAgent
 
 
 async def call_llm(llm_with_tools: ChatOpenAI, state: MessagesState):
@@ -18,7 +18,7 @@ async def call_llm(llm_with_tools: ChatOpenAI, state: MessagesState):
 
 
 async def main():
-    mcp_client = MultiServerMCPClient(
+    mcp_client_trello = MultiServerMCPClient(
         {
             "trello": {
                 "url": "http://localhost:8000/sse",
@@ -27,10 +27,21 @@ async def main():
         }
     )
 
+    mcp_client_github = MultiServerMCPClient(
+        {
+            "github": {
+                "url": "https://callecaje-git-repo-explorer.hf.space/gradio_api/mcp/sse",
+                "transport": "sse",
+            }
+        }
+    )
+
     memory = MemorySaver()
 
-    tools = await mcp_client.get_tools()
-    tool_node = ToolNode(tools)
+    trello_tools = await mcp_client_trello.get_tools()
+    github_tools = await mcp_client_github.get_tools()
+    tool_node = ToolNode(github_tools + trello_tools)
+
     llm = ChatOpenAI(
         model="meta-llama/Meta-Llama-3.1-8B-Instruct",
         temperature=0.0,
@@ -38,14 +49,14 @@ async def main():
         base_url="https://api.studio.nebius.com/v1/",
     )
 
-    trello_agent = TrelloAgent(
-        tools=tools,
+    trello_agent = ProjectManagerAgent(
+        tools=github_tools + trello_tools,
         llm=llm,
     )
 
     graph = StateGraph(MessagesState)
 
-    graph.add_node("llm", trello_agent.acall_trello_agent)
+    graph.add_node("llm", trello_agent.acall_pm_agent)
     graph.add_node("tool", tool_node)
     graph.set_entry_point("llm")
 

pmcp/agents/pm_agent.py

@@ -0,0 +1,36 @@
+from typing import List
+
+from pmcp.agents.agent_base import AgentBlueprint
+from langchain_core.tools import BaseTool
+from langchain_core.messages import SystemMessage
+from langchain_openai import ChatOpenAI
+from langgraph.graph import MessagesState
+
+SYSTEM_PROMPT = """
+You are an assistant that can manage Trello boards and projects and also look at github in order to add info on some trello cards.
+You will be given a set of tools to work with. Each time you decide to use a tool that modifies in any way a Trello board, you MUST ask the user if wants to proceed.
+If the user's answer is negative, then you have to abort everything and end the conversation.
+"""
+
+
+class ProjectManagerAgent:
+    def __init__(self, tools: List[BaseTool], llm: ChatOpenAI):
+        self.agent = AgentBlueprint(
+            agent_name="PM_AGENT",
+            description="The agent that performs actions on Trello and Github",
+            tools=tools,
+            system_prompt=SYSTEM_PROMPT.strip(),
+            llm=llm,
+        )
+
+    def call_pm_agent(self, state: MessagesState):
+        response = self.agent.call_agent(
+            [SystemMessage(content=self.agent.system_prompt)] + state.get("messages")
+        )
+        return {"messages": [response]}
+
+    async def acall_pm_agent(self, state: MessagesState):
+        response = await self.agent.acall_agent(
+            [SystemMessage(content=self.agent.system_prompt)] + state.get("messages")
+        )
+        return {"messages": [response]}

pmcp/agents/trello_agent.py

@@ -15,7 +15,7 @@ If the user's answer is negative, then you have to abort everything and end the
 
 class TrelloAgent:
     def __init__(self, tools: List[BaseTool], llm: ChatOpenAI):
-        self.__agent = AgentBlueprint(
+        self.agent = AgentBlueprint(
             agent_name="TRELLO_AGENT",
             description="The agent that performs actions on Trello",
             tools=tools,
@@ -24,13 +24,13 @@ class TrelloAgent:
         )
 
     def call_trello_agent(self, state: MessagesState):
-        response = self.__agent.call_agent(
-            [SystemMessage(content=self.__agent.system_prompt)] + state.get("messages")
+        response = self.agent.call_agent(
+            [SystemMessage(content=self.agent.system_prompt)] + state.get("messages")
         )
         return {"messages": [response]}
 
     async def acall_trello_agent(self, state: MessagesState):
-        response = await self.__agent.acall_agent(
-            [SystemMessage(content=self.__agent.system_prompt)] + state.get("messages")
+        response = await self.agent.acall_agent(
+            [SystemMessage(content=self.agent.system_prompt)] + state.get("messages")
         )
         return {"messages": [response]}