github-agent 0.2.2__tar.gz → 0.2.5__tar.gz

{github_agent-0.2.2 → github_agent-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 0.2.2
+Version: 0.2.5
 Summary: GitHub Agent for MCP
 Author-email: Audel Rouhi <knucklessg1@gmail.com>
 License: MIT
@@ -12,8 +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,huggingface,openai,web]>=1.32.0
-Requires-Dist: pydantic-ai-skills
+Requires-Dist: pydantic-ai-slim[a2a,ag-ui,anthropic,fastmcp,google,groq,huggingface,mistral,openai,web]>=1.58.0
+Requires-Dist: pydantic-ai-skills>=v0.4.0
 Requires-Dist: fastapi>=0.128.0
 Requires-Dist: fastmcp
 Requires-Dist: uvicorn
@@ -43,7 +43,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.2*
+*Version: 0.2.5*
 
 ## Overview
 

{github_agent-0.2.2 → github_agent-0.2.5}/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.2*
+*Version: 0.2.5*
 
 ## Overview
 

{github_agent-0.2.2 → github_agent-0.2.5}/github_agent/github_agent.py

@@ -7,6 +7,7 @@ import os
 import argparse
 import logging
 import uvicorn
+import httpx
 from typing import Optional, Any
 from contextlib import asynccontextmanager
 
@@ -38,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.2"
+__version__ = "0.2.5"
 
 logging.basicConfig(
     level=logging.INFO,
@@ -63,7 +64,6 @@ DEFAULT_SKILLS_DIRECTORY = os.getenv("SKILLS_DIRECTORY", get_skills_path())
 DEFAULT_ENABLE_WEB_UI = to_boolean(os.getenv("ENABLE_WEB_UI", "False"))
 DEFAULT_SSL_VERIFY = to_boolean(os.getenv("SSL_VERIFY", "True"))
 
-# Model Settings
 DEFAULT_MAX_TOKENS = to_integer(os.getenv("MAX_TOKENS", "16384"))
 DEFAULT_TEMPERATURE = to_float(os.getenv("TEMPERATURE", "0.7"))
 DEFAULT_TOP_P = to_float(os.getenv("TOP_P", "1.0"))
@@ -83,9 +83,6 @@ AGENT_DESCRIPTION = (
     "A multi-agent system for interacting with GitHub via delegated specialists."
 )
 
-# -------------------------------------------------------------------------
-# 1. System Prompts
-# -------------------------------------------------------------------------
 
 SUPERVISOR_SYSTEM_PROMPT = os.environ.get(
     "SUPERVISOR_SYSTEM_PROMPT",
@@ -264,16 +261,12 @@ SUPPORT_DOCS_AGENT_PROMPT = os.environ.get(
     ),
 )
 
-# -------------------------------------------------------------------------
-# 2. Agent Creation Logic
-# -------------------------------------------------------------------------
-
 
 def create_agent(
     provider: str = DEFAULT_PROVIDER,
     model_id: str = DEFAULT_MODEL_ID,
-    base_url: Optional[str] = None,
-    api_key: Optional[str] = None,
+    base_url: Optional[str] = DEFAULT_LLM_BASE_URL,
+    api_key: Optional[str] = DEFAULT_LLM_API_KEY,
     mcp_url: str = DEFAULT_MCP_URL,
     mcp_config: str = DEFAULT_MCP_CONFIG,
     skills_directory: Optional[str] = DEFAULT_SKILLS_DIRECTORY,
@@ -306,25 +299,29 @@ def create_agent(
         extra_body=DEFAULT_EXTRA_BODY,
     )
 
-    # Load master toolsets
     agent_toolsets = []
     if mcp_url:
         if "sse" in mcp_url.lower():
-            server = MCPServerSSE(mcp_url)
+            server = MCPServerSSE(
+                mcp_url, http_client=httpx.AsyncClient(verify=ssl_verify)
+            )
         else:
-            server = MCPServerStreamableHTTP(mcp_url)
+            server = MCPServerStreamableHTTP(
+                mcp_url, http_client=httpx.AsyncClient(verify=ssl_verify)
+            )
         agent_toolsets.append(server)
         logger.info(f"Connected to MCP Server: {mcp_url}")
     elif mcp_config:
         mcp_toolset = load_mcp_servers(mcp_config)
+        for server in mcp_toolset:
+            if hasattr(server, "http_client"):
+                server.http_client = httpx.AsyncClient(verify=ssl_verify)
         agent_toolsets.extend(mcp_toolset)
         logger.info(f"Connected to MCP Config JSON: {mcp_toolset}")
 
     if skills_directory and os.path.exists(skills_directory):
         agent_toolsets.append(SkillsToolset(directories=[str(skills_directory)]))
 
-    # Define Tag -> Prompt map
-    # Key is the MCP Tool Tag (or set of tags), Value is (SystemPrompt, AgentName)
     agent_defs = {
         "person": (CONTEXT_AGENT_PROMPT, "GitHub_Context_Agent"),
         "workflow": (ACTIONS_AGENT_PROMPT, "GitHub_Actions_Agent"),
@@ -383,7 +380,6 @@ def create_agent(
         )
         child_agents[tag] = agent
 
-    # Create Supervisor
     supervisor = Agent(
         name=AGENT_NAME,
         system_prompt=SUPERVISOR_SYSTEM_PROMPT,
@@ -392,9 +388,6 @@ def create_agent(
         deps_type=Any,
     )
 
-    # Define delegation tools
-    # We define these explicitly to give the Supervisor clear, typed tools.
-
     @supervisor.tool
     async def assign_task_to_context_agent(ctx: RunContext[Any], task: str) -> str:
         """Assign a task related to user context and general GitHub status to the Context Agent."""
@@ -570,8 +563,8 @@ def create_agent(
 def create_agent_server(
     provider: str = DEFAULT_PROVIDER,
     model_id: str = DEFAULT_MODEL_ID,
-    base_url: Optional[str] = None,
-    api_key: Optional[str] = None,
+    base_url: Optional[str] = DEFAULT_LLM_BASE_URL,
+    api_key: Optional[str] = DEFAULT_LLM_API_KEY,
     mcp_url: str = DEFAULT_MCP_URL,
     mcp_config: str = DEFAULT_MCP_CONFIG,
     skills_directory: Optional[str] = DEFAULT_SKILLS_DIRECTORY,
@@ -582,7 +575,12 @@ def create_agent_server(
     ssl_verify: bool = DEFAULT_SSL_VERIFY,
 ):
     print(
-        f"Starting {AGENT_NAME} with provider={provider}, model={model_id}, mcp={mcp_url} | {mcp_config}"
+        f"Starting {AGENT_NAME}:"
+        f"\tprovider={provider}"
+        f"\tmodel={model_id}"
+        f"\tbase_url={base_url}"
+        f"\tmcp={mcp_url} | {mcp_config}"
+        f"\tssl_verify={ssl_verify}"
     )
     agent = create_agent(
         provider=provider,
@@ -651,7 +649,6 @@ def create_agent_server(
                 status_code=422,
             )
 
-        # Prune large messages from history
         if hasattr(run_input, "messages"):
             run_input.messages = prune_large_messages(run_input.messages)
 

{github_agent-0.2.2 → github_agent-0.2.5}/github_agent/utils.py

@@ -3,7 +3,6 @@
 
 import os
 
-# Try importing specific clients/providers, but don't fail if variables are missing
 try:
 
     from openai import AsyncOpenAI
@@ -128,22 +127,17 @@ def prune_large_messages(messages: list[Any], max_length: int = 5000) -> list[An
                 f"... {content[-200:]}"
             )
 
-            # Replace content
             if isinstance(msg, dict):
                 msg["content"] = summary
                 pruned_messages.append(msg)
             elif hasattr(msg, "content"):
-                # Try to create a copy or modify in place if mutable
-                # If it's a Pydantic model it might be immutable or require copy
                 try:
-                    # Attempt shallow copy with update
                     from copy import copy
 
                     new_msg = copy(msg)
                     new_msg.content = summary
                     pruned_messages.append(new_msg)
                 except Exception:
-                    # Fallback: keep original if we can't modify
                     pruned_messages.append(msg)
             else:
                 pruned_messages.append(msg)
@@ -218,7 +212,6 @@ def load_skills_from_directory(directory: str) -> List[Skill]:
             if skill_file.exists():
                 try:
                     with open(skill_file, "r") as f:
-                        # Extract frontmatter
                         content = f.read()
                         if content.startswith("---"):
                             _, frontmatter, _ = content.split("---", 2)
@@ -254,35 +247,47 @@ def get_http_client(ssl_verify: bool = True) -> httpx.AsyncClient | None:
 def create_model(
     provider: str,
     model_id: str,
-    base_url: Optional[str] = None,
-    api_key: Optional[str] = None,
+    base_url: Optional[str],
+    api_key: Optional[str],
     ssl_verify: bool = True,
 ):
-    http_client = get_http_client(ssl_verify=ssl_verify)
+    """
+    Create a Pydantic AI model with the specified provider and configuration.
+
+    Args:
+        provider: The model provider (openai, anthropic, google, groq, mistral, huggingface, ollama)
+        model_id: The specific model ID to use
+        base_url: Optional base URL for the API
+        api_key: Optional API key
+        ssl_verify: Whether to verify SSL certificates (default: True)
+
+    Returns:
+        A Pydantic AI Model instance
+    """
+    http_client = None
+    if not ssl_verify:
+        http_client = httpx.AsyncClient(verify=False)
 
     if provider == "openai":
         target_base_url = base_url
         target_api_key = api_key
 
-        # If we have a custom client or specific settings, we might want to use the explicit provider object
         if http_client and AsyncOpenAI and OpenAIProvider:
             client = AsyncOpenAI(
-                api_key=target_api_key or os.environ.get("LLM_API_KEY"),
-                base_url=target_base_url or os.environ.get("LLM_BASE_URL"),
+                api_key=target_api_key or os.environ.get("OPENAI_API_KEY"),
+                base_url=target_base_url or os.environ.get("OPENAI_BASE_URL"),
                 http_client=http_client,
             )
             provider_instance = OpenAIProvider(openai_client=client)
             return OpenAIChatModel(model_name=model_id, provider=provider_instance)
 
-        # Fallback to standard env vars
         if target_base_url:
-            os.environ["LLM_BASE_URL"] = target_base_url
+            os.environ["OPENAI_BASE_URL"] = target_base_url
         if target_api_key:
-            os.environ["LLM_API_KEY"] = target_api_key
+            os.environ["OPENAI_API_KEY"] = target_api_key
         return OpenAIChatModel(model_name=model_id, provider="openai")
 
     elif provider == "ollama":
-        # Ollama is OpenAI compatible
         target_base_url = base_url or "http://localhost:11434/v1"
         target_api_key = api_key or "ollama"
 
@@ -295,18 +300,18 @@ def create_model(
             provider_instance = OpenAIProvider(openai_client=client)
             return OpenAIChatModel(model_name=model_id, provider=provider_instance)
 
-        os.environ["LLM_BASE_URL"] = target_base_url
-        os.environ["LLM_API_KEY"] = target_api_key
+        os.environ["OPENAI_BASE_URL"] = target_base_url
+        os.environ["OPENAI_API_KEY"] = target_api_key
         return OpenAIChatModel(model_name=model_id, provider="openai")
 
     elif provider == "anthropic":
         if api_key:
-            os.environ["LLM_API_KEY"] = api_key
+            os.environ["ANTHROPIC_API_KEY"] = api_key
 
         try:
             if http_client and AsyncAnthropic and AnthropicProvider:
                 client = AsyncAnthropic(
-                    api_key=api_key or os.environ.get("LLM_API_KEY"),
+                    api_key=api_key or os.environ.get("ANTHROPIC_API_KEY"),
                     http_client=http_client,
                 )
                 provider_instance = AnthropicProvider(anthropic_client=client)
@@ -317,18 +322,17 @@ def create_model(
         return AnthropicModel(model_name=model_id)
 
     elif provider == "google":
-        # Google generic setup, skipping complex SSL for now as agreed
         if api_key:
-            os.environ["LLM_API_KEY"] = api_key
+            os.environ["GEMINI_API_KEY"] = api_key
         return GoogleModel(model_name=model_id)
 
     elif provider == "groq":
         if api_key:
-            os.environ["LLM_API_KEY"] = api_key
+            os.environ["GROQ_API_KEY"] = api_key
 
         if http_client and AsyncGroq and GroqProvider:
             client = AsyncGroq(
-                api_key=api_key or os.environ.get("LLM_API_KEY"),
+                api_key=api_key or os.environ.get("GROQ_API_KEY"),
                 http_client=http_client,
             )
             provider_instance = GroqProvider(groq_client=client)
@@ -338,18 +342,16 @@ def create_model(
 
     elif provider == "mistral":
         if api_key:
-            os.environ["LLM_API_KEY"] = api_key
+            os.environ["MISTRAL_API_KEY"] = api_key
 
         if http_client and Mistral and MistralProvider:
-            # Assuming mistral_client argument for MistralProvider
-            # Ideally we would verify this, but we'll try standard pattern
             pass
 
         return MistralModel(model_name=model_id)
 
     elif provider == "huggingface":
         if api_key:
-            os.environ["LLM_API_KEY"] = api_key
+            os.environ["HUGGING_FACE_API_KEY"] = api_key
         return HuggingFaceModel(model_name=model_id)
 
     return OpenAIChatModel(model_name=model_id, provider="openai")

{github_agent-0.2.2 → github_agent-0.2.5}/github_agent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 0.2.2
+Version: 0.2.5
 Summary: GitHub Agent for MCP
 Author-email: Audel Rouhi <knucklessg1@gmail.com>
 License: MIT
@@ -12,8 +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,huggingface,openai,web]>=1.32.0
-Requires-Dist: pydantic-ai-skills
+Requires-Dist: pydantic-ai-slim[a2a,ag-ui,anthropic,fastmcp,google,groq,huggingface,mistral,openai,web]>=1.58.0
+Requires-Dist: pydantic-ai-skills>=v0.4.0
 Requires-Dist: fastapi>=0.128.0
 Requires-Dist: fastmcp
 Requires-Dist: uvicorn
@@ -43,7 +43,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.2*
+*Version: 0.2.5*
 
 ## Overview
 

github_agent-0.2.5/github_agent.egg-info/requires.txt

@@ -0,0 +1,6 @@
+pydantic-ai-slim[a2a,ag-ui,anthropic,fastmcp,google,groq,huggingface,mistral,openai,web]>=1.58.0
+pydantic-ai-skills>=v0.4.0
+fastapi>=0.128.0
+fastmcp
+uvicorn
+fastapi

{github_agent-0.2.2 → github_agent-0.2.5}/pyproject.toml

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

github_agent-0.2.2/github_agent.egg-info/requires.txt

@@ -1,6 +0,0 @@
-pydantic-ai-slim[a2a,ag-ui,anthropic,fastmcp,google,huggingface,openai,web]>=1.32.0
-pydantic-ai-skills
-fastapi>=0.128.0
-fastmcp
-uvicorn
-fastapi