github-agent 0.2.12__tar.gz → 0.2.14__tar.gz

{github_agent-0.2.12 → github_agent-0.2.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 0.2.12
+Version: 0.2.14
 Summary: GitHub Agent for MCP
 Author-email: Audel Rouhi <knucklessg1@gmail.com>
 License: MIT
@@ -12,7 +12,8 @@ Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: pydantic-ai-slim[a2a,ag-ui,anthropic,fastmcp,google,groq,huggingface,mistral,openai,web]>=1.58.0
+Requires-Dist: tree-sitter>=0.23.2
+Requires-Dist: pydantic-ai-slim[a2a,ag-ui,anthropic,fastmcp,google,groq,huggingface,mistral,openai,web]>=1.60.0
 Requires-Dist: pydantic-ai-skills>=v0.4.0
 Requires-Dist: fastapi>=0.128.0
 Requires-Dist: fastmcp
@@ -43,7 +44,7 @@ Dynamic: license-file
 ![PyPI - Wheel](https://img.shields.io/pypi/wheel/github-agent)
 ![PyPI - Implementation](https://img.shields.io/pypi/implementation/github-agent)
 
-*Version: 0.2.12*
+*Version: 0.2.14*
 
 ## Overview
 

{github_agent-0.2.12 → github_agent-0.2.14}/README.md

@@ -21,7 +21,7 @@
 ![PyPI - Wheel](https://img.shields.io/pypi/wheel/github-agent)
 ![PyPI - Implementation](https://img.shields.io/pypi/implementation/github-agent)
 
-*Version: 0.2.12*
+*Version: 0.2.14*
 
 ## Overview
 

{github_agent-0.2.12 → github_agent-0.2.14}/github_agent/github_agent.py

@@ -39,7 +39,7 @@ from pydantic import ValidationError
 from pydantic_ai.ui import SSE_CONTENT_TYPE
 from pydantic_ai.ui.ag_ui import AGUIAdapter
 
-__version__ = "0.2.12"
+__version__ = "0.2.14"
 
 logging.basicConfig(
     level=logging.INFO,
@@ -363,7 +363,23 @@ def create_agent(
         ),
     }
 
+    # 1. Identify Universal Skills
+    # Universal skills are those in the skills directory that do NOT start with the package prefix
+    package_prefix = "github-"
+    skills_path = get_skills_path()
+    universal_skill_dirs = []
+
+    if os.path.exists(skills_path):
+        for item in os.listdir(skills_path):
+            item_path = os.path.join(skills_path, item)
+            if os.path.isdir(item_path):
+                if not item.startswith(package_prefix):
+                    universal_skill_dirs.append(item_path)
+                    logger.info(f"Identified universal skill: {item}")
+
+    supervisor_skills = []
     child_agents = {}
+    supervisor_skills_directories = [get_skills_path()]
 
     for tag, (system_prompt, agent_name) in agent_defs.items():
         tag_toolsets = []
@@ -381,21 +397,28 @@ def create_agent(
         # Load specific skills for this tag
         skill_dir_name = f"github-{tag.replace('_', '-')}"
 
+        child_skills_directories = []
+
         # Check custom skills directory
         if custom_skills_directory:
             skill_dir_path = os.path.join(custom_skills_directory, skill_dir_name)
             if os.path.exists(skill_dir_path):
-                tag_toolsets.append(SkillsToolset(directories=[skill_dir_path]))
-                logger.info(
-                    f"Loaded specialized skills for {tag} from {skill_dir_path}"
-                )
+                child_skills_directories.append(skill_dir_path)
 
         # Check default skills directory
         default_skill_path = os.path.join(get_skills_path(), skill_dir_name)
         if os.path.exists(default_skill_path):
-            tag_toolsets.append(SkillsToolset(directories=[default_skill_path]))
+            child_skills_directories.append(default_skill_path)
+
+        # Append Universal Skills to ALL child agents
+        if universal_skill_dirs:
+            child_skills_directories.extend(universal_skill_dirs)
+
+        if child_skills_directories:
+            ts = SkillsToolset(directories=child_skills_directories)
+            tag_toolsets.append(ts)
             logger.info(
-                f"Loaded specialized skills for {tag} from {default_skill_path}"
+                f"Loaded specialized skills for {tag} from {child_skills_directories}"
             )
 
         # Collect tool names for logging
@@ -434,11 +457,43 @@ def create_agent(
         )
         child_agents[tag] = agent
 
+    # Create Custom Agent if custom_skills_directory is provided
+    if custom_skills_directory:
+        custom_agent_tag = "custom_agent"
+        custom_agent_name = "Custom_Agent"
+        custom_agent_prompt = (
+            "You are the Custom Agent.\n"
+            "Your goal is to handle custom tasks or general tasks not covered by other specialists.\n"
+            "You have access to valid custom skills and universal skills."
+        )
+
+        custom_agent_skills_dirs = list(universal_skill_dirs)
+        custom_agent_skills_dirs.append(custom_skills_directory)
+
+        custom_toolsets = []
+        custom_toolsets.append(SkillsToolset(directories=custom_agent_skills_dirs))
+
+        custom_agent = Agent(
+            name=custom_agent_name,
+            system_prompt=custom_agent_prompt,
+            model=model,
+            model_settings=settings,
+            toolsets=custom_toolsets,
+            tool_timeout=DEFAULT_TOOL_TIMEOUT,
+        )
+        child_agents[custom_agent_tag] = custom_agent
+
+    if custom_skills_directory:
+        supervisor_skills_directories.append(custom_skills_directory)
+    supervisor_skills.append(SkillsToolset(directories=supervisor_skills_directories))
+    logger.info(f"Loaded supervisor skills from: {supervisor_skills_directories}")
+
     supervisor = Agent(
         name=AGENT_NAME,
         system_prompt=SUPERVISOR_SYSTEM_PROMPT,
         model=model,
         model_settings=settings,
+        toolsets=supervisor_skills,
         deps_type=Any,
     )
 
@@ -707,6 +762,17 @@ def create_agent(
             logger.exception(f"Error in Support Docs Agent: {e}")
             return f"Error executing task for Support Docs Agent: {e}"
 
+    if "custom_agent" in child_agents:
+
+        @supervisor.tool
+        async def assign_task_to_custom_agent(ctx: RunContext[Any], task: str) -> str:
+            """Assign a task to the Custom Agent."""
+            return (
+                await child_agents["custom_agent"]
+                .run(task, usage=ctx.usage, deps=ctx.deps)
+                .output
+            )
+
     return supervisor
 
 

github_agent-0.2.14/github_agent/skills/a2a_client/scripts/a2a_client.py

@@ -0,0 +1,212 @@
+#!/usr/bin/env python3
+import asyncio
+import httpx
+import json
+import uuid
+import argparse
+import sys
+
+
+async def validate_agent_card(client, agent_url):
+    """
+    Validates the agent by fetching its well-known agent card.
+    """
+    card_url = f"{agent_url.rstrip('/')}/.well-known/agent-card.json"
+    print(f"Fetching agent card from: {card_url}")
+    try:
+        resp = await client.get(card_url)
+        if resp.status_code == 200:
+            try:
+                card_data = resp.json()
+                print(f"Agent Card Found: {json.dumps(card_data, indent=2)}")
+                return True
+            except json.JSONDecodeError:
+                print(f"Failed to decode agent card JSON from {card_url}")
+                return False
+        else:
+            print(f"Failed to fetch agent card. Status Code: {resp.status_code}")
+            return False
+    except httpx.RequestError as e:
+        print(f"Connection failed to {card_url}: {e}")
+        return False
+
+
+async def send_message(client, agent_url, message_text):
+    """
+    Sends a message to the agent via JSON-RPC.
+    """
+    print(f"\nSending Message: '{message_text}' to {agent_url}")
+
+    payload = {
+        "jsonrpc": "2.0",
+        "method": "message/send",
+        "params": {
+            "message": {
+                "kind": "message",
+                "role": "user",
+                "parts": [{"kind": "text", "text": message_text}],
+                "messageId": str(uuid.uuid4()),
+            }
+        },
+        "id": 1,
+    }
+
+    try:
+        resp = await client.post(
+            agent_url, json=payload, headers={"Content-Type": "application/json"}
+        )
+
+        if resp.status_code != 200:
+            print(f"Error sending message. Status Code: {resp.status_code}")
+            print(resp.text)
+            return None
+
+        data = resp.json()
+        if "error" in data:
+            print(f"JSON-RPC Error: {data['error']}")
+            return None
+
+        if "result" in data and "id" in data["result"]:
+            task_id = data["result"]["id"]
+            print(f"Task Submitted with ID: {task_id}")
+            return task_id
+        else:
+            print(f"Unexpected response format: {data}")
+            return None
+
+    except httpx.RequestError as e:
+        print(f"Connection failed during message send: {e}")
+        return None
+    except json.JSONDecodeError:
+        print(f"Failed to decode response JSON: {resp.text}")
+        return None
+
+
+async def poll_task(client, agent_url, task_id):
+    """
+    Polls the task status until completion.
+    """
+    print(f"Polling for result for Task ID: {task_id}...")
+
+    while True:
+        await asyncio.sleep(2)
+        poll_payload = {
+            "jsonrpc": "2.0",
+            "method": "tasks/get",
+            "params": {"id": task_id},
+            "id": 2,
+        }
+
+        try:
+            poll_resp = await client.post(
+                agent_url,
+                json=poll_payload,
+                headers={"Content-Type": "application/json"},
+            )
+
+            if poll_resp.status_code != 200:
+                print(f"Polling Failed: {poll_resp.status_code}")
+                print(f"Details: {poll_resp.text}")
+                break
+
+            poll_data = poll_resp.json()
+
+            if "error" in poll_data:
+                print(f"Polling Error: {poll_data['error']}")
+                break
+
+            if "result" in poll_data:
+                status = poll_data["result"].get("status", {})
+                state = status.get("state")
+                print(f"Task State: {state}")
+
+                if state not in ["submitted", "running", "working"]:
+                    print(f"\nTask Finished with state: {state}")
+                    return poll_data["result"]
+            else:
+                print(f"Unexpected polling response: {poll_data}")
+                break
+
+        except httpx.RequestError as e:
+            print(f"Connection failed during polling: {e}")
+            break
+        except json.JSONDecodeError:
+            print(f"Failed to decode polling response: {poll_resp.text}")
+            break
+
+
+def print_result(result):
+    """
+    Prints the final result from the agent.
+    """
+    if not result:
+        return
+
+    history = result.get("history", [])
+    if history:
+        last_msg = None
+        # Find the last message that is NOT from the user (i.e., the agent's response)
+        for msg in reversed(history):
+            if msg.get("role") != "user":
+                last_msg = msg
+                break
+
+        if last_msg:
+            print("\n--- Agent Response ---")
+            if "parts" in last_msg:
+                for part in last_msg["parts"]:
+                    if "text" in part:
+                        print(part["text"])
+                    elif "content" in part:
+                        print(part["content"])
+            else:
+                print(f"Final Message (No parts): {last_msg}")
+        else:
+            print("\n--- No Agent Response Found in History ---")
+
+    # print(f"\nFull Result Debug:\n{json.dumps(result, indent=2)}")
+
+
+async def main():
+    parser = argparse.ArgumentParser(
+        description="A2A Client for communicating with other agents."
+    )
+    parser.add_argument(
+        "--url",
+        required=True,
+        help="The base URL of the A2A Agent (e.g., http://agent.arpa/a2a/)",
+    )
+    parser.add_argument(
+        "--query", required=True, help="The message/query to send to the agent"
+    )
+
+    args = parser.parse_args()
+
+    agent_url = args.url
+    query = args.query
+
+    print("Initializing A2A Client...")
+    print(f"Target Agent: {agent_url}")
+    print(f"Query: {query}")
+
+    async with httpx.AsyncClient(timeout=60.0) as client:
+        # 1. Validate Agent
+        if not await validate_agent_card(client, agent_url):
+            print("Agent validation failed. Aborting.")
+            sys.exit(1)
+
+        # 2. Send Message
+        task_id = await send_message(client, agent_url, query)
+        if not task_id:
+            print("Failed to submit task. Aborting.")
+            sys.exit(1)
+
+        # 3. Poll for Result
+        result = await poll_task(client, agent_url, task_id)
+
+        # 4. Print Result
+        print_result(result)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

{github_agent-0.2.12 → github_agent-0.2.14}/github_agent/utils.py

@@ -2,9 +2,22 @@
 # coding: utf-8
 
 import os
+import httpx
+import pickle
+import yaml
+import logging
+from pathlib import Path
+from typing import Union, List, Any, Optional
+import json
+from importlib.resources import files, as_file
+from pydantic_ai.models.openai import OpenAIChatModel
+from pydantic_ai.models.google import GoogleModel
+from pydantic_ai.models.huggingface import HuggingFaceModel
+from pydantic_ai.models.groq import GroqModel
+from pydantic_ai.models.mistral import MistralModel
+from fasta2a import Skill
 
 try:
-
     from openai import AsyncOpenAI
     from pydantic_ai.providers.openai import OpenAIProvider
 except ImportError:
@@ -26,26 +39,16 @@ except ImportError:
     MistralProvider = None
 
 try:
+    from pydantic_ai.models.anthropic import AnthropicModel
     from anthropic import AsyncAnthropic
     from pydantic_ai.providers.anthropic import AnthropicProvider
 except ImportError:
+    AnthropicModel = None
     AsyncAnthropic = None
     AnthropicProvider = None
 
-import httpx
-import pickle
-import yaml
-from pathlib import Path
-from typing import Any, Union, List, Optional
-import json
-from importlib.resources import files, as_file
-from pydantic_ai.models.openai import OpenAIChatModel
-from pydantic_ai.models.anthropic import AnthropicModel
-from pydantic_ai.models.google import GoogleModel
-from pydantic_ai.models.huggingface import HuggingFaceModel
-from pydantic_ai.models.groq import GroqModel
-from pydantic_ai.models.mistral import MistralModel
-from pydantic_ai_skills import Skill
+
+logger = logging.getLogger(__name__)
 
 
 def to_integer(string: Union[str, int] = None) -> int:
@@ -163,6 +166,9 @@ def load_model(file: str) -> Any:
 def retrieve_package_name() -> str:
     """
     Returns the top-level package name of the module that imported this utils.py.
+
+    Works reliably when utils.py is inside a proper package (with __init__.py or
+    implicit namespace package) and the caller does normal imports.
     """
     if __package__:
         top = __package__.partition(".")[0]
@@ -363,6 +369,16 @@ def create_model(
 def extract_tool_tags(tool_def: Any) -> List[str]:
     """
     Extracts tags from a tool definition object.
+
+    Found structure in debug:
+    tool_def.name (str)
+    tool_def.meta (dict) -> {'fastmcp': {'tags': ['tag']}}
+
+    This function checks multiple paths to be robust:
+    1. tool_def.meta['fastmcp']['tags']
+    2. tool_def.meta['tags']
+    3. tool_def.metadata['tags'] (legacy/alternative wrapper)
+    4. tool_def.metadata.get('meta')... (nested path)
     """
     tags_list = []
 
@@ -409,3 +425,10 @@ def tool_in_tag(tool_def: Any, tag: str) -> bool:
         return True
     else:
         return False
+
+
+def filter_tools_by_tag(tools: List[Any], tag: str) -> List[Any]:
+    """
+    Filters a list of tools for a given tag.
+    """
+    return [t for t in tools if tool_in_tag(t, tag)]

{github_agent-0.2.12 → github_agent-0.2.14}/github_agent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 0.2.12
+Version: 0.2.14
 Summary: GitHub Agent for MCP
 Author-email: Audel Rouhi <knucklessg1@gmail.com>
 License: MIT
@@ -12,7 +12,8 @@ Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: pydantic-ai-slim[a2a,ag-ui,anthropic,fastmcp,google,groq,huggingface,mistral,openai,web]>=1.58.0
+Requires-Dist: tree-sitter>=0.23.2
+Requires-Dist: pydantic-ai-slim[a2a,ag-ui,anthropic,fastmcp,google,groq,huggingface,mistral,openai,web]>=1.60.0
 Requires-Dist: pydantic-ai-skills>=v0.4.0
 Requires-Dist: fastapi>=0.128.0
 Requires-Dist: fastmcp
@@ -43,7 +44,7 @@ Dynamic: license-file
 ![PyPI - Wheel](https://img.shields.io/pypi/wheel/github-agent)
 ![PyPI - Implementation](https://img.shields.io/pypi/implementation/github-agent)
 
-*Version: 0.2.12*
+*Version: 0.2.14*
 
 ## Overview
 

{github_agent-0.2.12 → github_agent-0.2.14}/github_agent.egg-info/SOURCES.txt

@@ -10,4 +10,5 @@ github_agent.egg-info/dependency_links.txt
 github_agent.egg-info/entry_points.txt
 github_agent.egg-info/requires.txt
 github_agent.egg-info/top_level.txt
+github_agent/skills/a2a_client/scripts/a2a_client.py
 scripts/validate_a2a_agent.py

{github_agent-0.2.12 → github_agent-0.2.14}/github_agent.egg-info/requires.txt

@@ -1,4 +1,5 @@
-pydantic-ai-slim[a2a,ag-ui,anthropic,fastmcp,google,groq,huggingface,mistral,openai,web]>=1.58.0
+tree-sitter>=0.23.2
+pydantic-ai-slim[a2a,ag-ui,anthropic,fastmcp,google,groq,huggingface,mistral,openai,web]>=1.60.0
 pydantic-ai-skills>=v0.4.0
 fastapi>=0.128.0
 fastmcp

{github_agent-0.2.12 → github_agent-0.2.14}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "github-agent"
-version = "0.2.12"
+version = "0.2.14"
 readme = "README.md"
 description = "GitHub Agent for MCP"
 requires-python = ">=3.10"
@@ -18,7 +18,8 @@ classifiers = [
     "Programming Language :: Python :: 3",
 ]
 dependencies = [
-    "pydantic-ai-slim[fastmcp,openai,anthropic,groq,mistral,google,huggingface,a2a,ag-ui,web]>=1.58.0",
+    "tree-sitter>=0.23.2",
+    "pydantic-ai-slim[fastmcp,openai,groq,anthropic,mistral,google,huggingface,a2a,ag-ui,web]>=1.60.0",
     "pydantic-ai-skills>=v0.4.0",
     "fastapi>=0.128.0",
     "fastmcp",