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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 0.2.0
+Version: 0.2.2
 Summary: GitHub Agent for MCP
 Author-email: Audel Rouhi <knucklessg1@gmail.com>
 License: MIT
@@ -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.0*
+*Version: 0.2.2*
 
 ## Overview
 
@@ -172,7 +172,7 @@ The `github-agent` command starts the server.
 | `--port` | Port to bind the server to | `9000` |
 | `--mcp-config` | Path to MCP configuration file | `mcp_config.json` |
 | `--provider` | LLM Provider (openai, anthropic, google, etc.) | `openai` |
-| `--model-id` | LLM Model ID | `qwen/qwen3-4b-2507` |
+| `--model-id` | LLM Model ID | `qwen/qwen3-coder-next` |
 
 ### Running the Agent Server
 
@@ -193,7 +193,7 @@ docker build -t github-agent .
 ```bash
 docker run -d \
   -p 9000:9000 \
-  -e OPENAI_API_KEY=sk-... \
+  -e LLM_API_KEY=sk-... \
   -e MCP_CONFIG=/app/mcp_config.json \
   knucklessg1/github-agent:latest
 ```
@@ -211,7 +211,7 @@ services:
     environment:
       - PROVIDER=openai
       - MODEL_ID=gpt-4o
-      - OPENAI_API_KEY=${OPENAI_API_KEY}
+      - LLM_API_KEY=${LLM_API_KEY}
     volumes:
       - ./mcp_config.json:/app/mcp_config.json
 ```

{github_agent-0.2.0 → github_agent-0.2.2}/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.0*
+*Version: 0.2.2*
 
 ## Overview
 
@@ -150,7 +150,7 @@ The `github-agent` command starts the server.
 | `--port` | Port to bind the server to | `9000` |
 | `--mcp-config` | Path to MCP configuration file | `mcp_config.json` |
 | `--provider` | LLM Provider (openai, anthropic, google, etc.) | `openai` |
-| `--model-id` | LLM Model ID | `qwen/qwen3-4b-2507` |
+| `--model-id` | LLM Model ID | `qwen/qwen3-coder-next` |
 
 ### Running the Agent Server
 
@@ -171,7 +171,7 @@ docker build -t github-agent .
 ```bash
 docker run -d \
   -p 9000:9000 \
-  -e OPENAI_API_KEY=sk-... \
+  -e LLM_API_KEY=sk-... \
   -e MCP_CONFIG=/app/mcp_config.json \
   knucklessg1/github-agent:latest
 ```
@@ -189,7 +189,7 @@ services:
     environment:
       - PROVIDER=openai
       - MODEL_ID=gpt-4o
-      - OPENAI_API_KEY=${OPENAI_API_KEY}
+      - LLM_API_KEY=${LLM_API_KEY}
     volumes:
       - ./mcp_config.json:/app/mcp_config.json
 ```

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

@@ -11,7 +11,11 @@ from typing import Optional, Any
 from contextlib import asynccontextmanager
 
 from pydantic_ai import Agent, ModelSettings, RunContext
-from pydantic_ai.mcp import load_mcp_servers, MCPServerStreamableHTTP, MCPServerSSE
+from pydantic_ai.mcp import (
+    load_mcp_servers,
+    MCPServerStreamableHTTP,
+    MCPServerSSE,
+)
 from pydantic_ai_skills import SkillsToolset
 from fasta2a import Skill
 from github_agent.utils import (
@@ -34,7 +38,7 @@ from pydantic import ValidationError
 from pydantic_ai.ui import SSE_CONTENT_TYPE
 from pydantic_ai.ui.ag_ui import AGUIAdapter
 
-__version__ = "0.2.0"
+__version__ = "0.2.2"
 
 logging.basicConfig(
     level=logging.INFO,
@@ -50,18 +54,17 @@ DEFAULT_HOST = os.getenv("HOST", "0.0.0.0")
 DEFAULT_PORT = to_integer(string=os.getenv("PORT", "9000"))
 DEFAULT_DEBUG = to_boolean(string=os.getenv("DEBUG", "False"))
 DEFAULT_PROVIDER = os.getenv("PROVIDER", "openai")
-DEFAULT_MODEL_ID = os.getenv("MODEL_ID", "qwen/qwen3-4b-2507")
-DEFAULT_OPENAI_BASE_URL = os.getenv(
-    "OPENAI_BASE_URL", "http://host.docker.internal:1234/v1"
-)
-DEFAULT_OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "ollama")
+DEFAULT_MODEL_ID = os.getenv("MODEL_ID", "qwen/qwen3-coder-next")
+DEFAULT_LLM_BASE_URL = os.getenv("LLM_BASE_URL", "http://host.docker.internal:1234/v1")
+DEFAULT_LLM_API_KEY = os.getenv("LLM_API_KEY", "ollama")
 DEFAULT_MCP_URL = os.getenv("MCP_URL", None)
 DEFAULT_MCP_CONFIG = os.getenv("MCP_CONFIG", get_mcp_config_path())
 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", "8192"))
+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"))
 DEFAULT_TIMEOUT = to_float(os.getenv("TIMEOUT", "32400.0"))
@@ -261,7 +264,6 @@ SUPPORT_DOCS_AGENT_PROMPT = os.environ.get(
     ),
 )
 
-
 # -------------------------------------------------------------------------
 # 2. Agent Creation Logic
 # -------------------------------------------------------------------------
@@ -275,13 +277,20 @@ def create_agent(
     mcp_url: str = DEFAULT_MCP_URL,
     mcp_config: str = DEFAULT_MCP_CONFIG,
     skills_directory: Optional[str] = DEFAULT_SKILLS_DIRECTORY,
+    ssl_verify: bool = DEFAULT_SSL_VERIFY,
 ) -> Agent:
     """
     Creates the Supervisor Agent with sub-agents registered as tools.
     """
     logger.info("Initializing Multi-Agent System for GitHub...")
 
-    model = create_model(provider, model_id, base_url, api_key)
+    model = create_model(
+        provider=provider,
+        model_id=model_id,
+        base_url=base_url,
+        api_key=api_key,
+        ssl_verify=ssl_verify,
+    )
     settings = ModelSettings(
         max_tokens=DEFAULT_MAX_TOKENS,
         temperature=DEFAULT_TEMPERATURE,
@@ -298,21 +307,21 @@ def create_agent(
     )
 
     # Load master toolsets
-    master_toolsets = []
-    if mcp_config:
-        mcp_toolset = load_mcp_servers(mcp_config)
-        master_toolsets.extend(mcp_toolset)
-        logger.info(f"Connected to MCP Config JSON: {mcp_toolset}")
-    elif mcp_url:
+    agent_toolsets = []
+    if mcp_url:
         if "sse" in mcp_url.lower():
             server = MCPServerSSE(mcp_url)
         else:
             server = MCPServerStreamableHTTP(mcp_url)
-        master_toolsets.append(server)
+        agent_toolsets.append(server)
         logger.info(f"Connected to MCP Server: {mcp_url}")
+    elif mcp_config:
+        mcp_toolset = load_mcp_servers(mcp_config)
+        agent_toolsets.extend(mcp_toolset)
+        logger.info(f"Connected to MCP Config JSON: {mcp_toolset}")
 
     if skills_directory and os.path.exists(skills_directory):
-        master_toolsets.append(SkillsToolset(directories=[str(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)
@@ -353,7 +362,7 @@ def create_agent(
 
     for tag, (system_prompt, agent_name) in agent_defs.items():
         tag_toolsets = []
-        for ts in master_toolsets:
+        for ts in agent_toolsets:
 
             def filter_func(ctx, tool_def, t=tag):
                 return tool_in_tag(tool_def, t)
@@ -570,6 +579,7 @@ def create_agent_server(
     host: Optional[str] = DEFAULT_HOST,
     port: Optional[int] = DEFAULT_PORT,
     enable_web_ui: bool = DEFAULT_ENABLE_WEB_UI,
+    ssl_verify: bool = DEFAULT_SSL_VERIFY,
 ):
     print(
         f"Starting {AGENT_NAME} with provider={provider}, model={model_id}, mcp={mcp_url} | {mcp_config}"
@@ -582,6 +592,7 @@ def create_agent_server(
         mcp_url=mcp_url,
         mcp_config=mcp_config,
         skills_directory=skills_directory,
+        ssl_verify=ssl_verify,
     )
 
     if skills_directory and os.path.exists(skills_directory):
@@ -609,7 +620,11 @@ def create_agent_server(
 
     @asynccontextmanager
     async def lifespan(app: FastAPI):
-        yield
+        if hasattr(a2a_app, "router") and hasattr(a2a_app.router, "lifespan_context"):
+            async with a2a_app.router.lifespan_context(a2a_app):
+                yield
+        else:
+            yield
 
     app = FastAPI(
         title=f"{AGENT_NAME} - A2A + AG-UI Server",
@@ -692,10 +707,10 @@ def agent_server():
     parser.add_argument("--model-id", default=DEFAULT_MODEL_ID, help="LLM Model ID")
     parser.add_argument(
         "--base-url",
-        default=DEFAULT_OPENAI_BASE_URL,
+        default=DEFAULT_LLM_BASE_URL,
         help="LLM Base URL (for OpenAI compatible providers)",
     )
-    parser.add_argument("--api-key", default=DEFAULT_OPENAI_API_KEY, help="LLM API Key")
+    parser.add_argument("--api-key", default=DEFAULT_LLM_API_KEY, help="LLM API Key")
     parser.add_argument("--mcp-url", default=DEFAULT_MCP_URL, help="MCP Server URL")
     parser.add_argument(
         "--mcp-config", default=DEFAULT_MCP_CONFIG, help="MCP Server Config"
@@ -750,6 +765,7 @@ def agent_server():
         host=args.host,
         port=args.port,
         enable_web_ui=args.web,
+        ssl_verify=not args.insecure,
     )
 
 

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

@@ -2,6 +2,38 @@
 # coding: utf-8
 
 import os
+
+# Try importing specific clients/providers, but don't fail if variables are missing
+try:
+
+    from openai import AsyncOpenAI
+    from pydantic_ai.providers.openai import OpenAIProvider
+except ImportError:
+    AsyncOpenAI = None
+    OpenAIProvider = None
+
+try:
+    from groq import AsyncGroq
+    from pydantic_ai.providers.groq import GroqProvider
+except ImportError:
+    AsyncGroq = None
+    GroqProvider = None
+
+try:
+    from mistralai import Mistral
+    from pydantic_ai.providers.mistral import MistralProvider
+except ImportError:
+    Mistral = None
+    MistralProvider = None
+
+try:
+    from anthropic import AsyncAnthropic
+    from pydantic_ai.providers.anthropic import AnthropicProvider
+except ImportError:
+    AsyncAnthropic = None
+    AnthropicProvider = None
+
+import httpx
 import pickle
 import yaml
 from pathlib import Path
@@ -12,6 +44,8 @@ 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
 
 
@@ -211,36 +245,113 @@ def load_skills_from_directory(directory: str) -> List[Skill]:
     return skills
 
 
+def get_http_client(ssl_verify: bool = True) -> httpx.AsyncClient | None:
+    if not ssl_verify:
+        return httpx.AsyncClient(verify=False)
+    return None
+
+
 def create_model(
     provider: str,
     model_id: str,
-    base_url: Optional[str],
-    api_key: Optional[str],
+    base_url: Optional[str] = None,
+    api_key: Optional[str] = None,
+    ssl_verify: bool = True,
 ):
+    http_client = get_http_client(ssl_verify=ssl_verify)
+
     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"),
+                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["OPENAI_BASE_URL"] = target_base_url
+            os.environ["LLM_BASE_URL"] = target_base_url
         if target_api_key:
-            os.environ["OPENAI_API_KEY"] = target_api_key
+            os.environ["LLM_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"
+
+        if http_client and AsyncOpenAI and OpenAIProvider:
+            client = AsyncOpenAI(
+                api_key=target_api_key,
+                base_url=target_base_url,
+                http_client=http_client,
+            )
+            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
         return OpenAIChatModel(model_name=model_id, provider="openai")
 
     elif provider == "anthropic":
         if api_key:
-            os.environ["ANTHROPIC_API_KEY"] = api_key
+            os.environ["LLM_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"),
+                    http_client=http_client,
+                )
+                provider_instance = AnthropicProvider(anthropic_client=client)
+                return AnthropicModel(model_name=model_id, provider=provider_instance)
+        except ImportError:
+            pass
+
         return AnthropicModel(model_name=model_id)
 
     elif provider == "google":
+        # Google generic setup, skipping complex SSL for now as agreed
         if api_key:
-            os.environ["GEMINI_API_KEY"] = api_key
-            os.environ["GOOGLE_API_KEY"] = api_key
+            os.environ["LLM_API_KEY"] = api_key
         return GoogleModel(model_name=model_id)
 
+    elif provider == "groq":
+        if api_key:
+            os.environ["LLM_API_KEY"] = api_key
+
+        if http_client and AsyncGroq and GroqProvider:
+            client = AsyncGroq(
+                api_key=api_key or os.environ.get("LLM_API_KEY"),
+                http_client=http_client,
+            )
+            provider_instance = GroqProvider(groq_client=client)
+            return GroqModel(model_name=model_id, provider=provider_instance)
+
+        return GroqModel(model_name=model_id)
+
+    elif provider == "mistral":
+        if api_key:
+            os.environ["LLM_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["HF_TOKEN"] = api_key
+            os.environ["LLM_API_KEY"] = api_key
         return HuggingFaceModel(model_name=model_id)
+
     return OpenAIChatModel(model_name=model_id, provider="openai")
 
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 0.2.0
+Version: 0.2.2
 Summary: GitHub Agent for MCP
 Author-email: Audel Rouhi <knucklessg1@gmail.com>
 License: MIT
@@ -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.0*
+*Version: 0.2.2*
 
 ## Overview
 
@@ -172,7 +172,7 @@ The `github-agent` command starts the server.
 | `--port` | Port to bind the server to | `9000` |
 | `--mcp-config` | Path to MCP configuration file | `mcp_config.json` |
 | `--provider` | LLM Provider (openai, anthropic, google, etc.) | `openai` |
-| `--model-id` | LLM Model ID | `qwen/qwen3-4b-2507` |
+| `--model-id` | LLM Model ID | `qwen/qwen3-coder-next` |
 
 ### Running the Agent Server
 
@@ -193,7 +193,7 @@ docker build -t github-agent .
 ```bash
 docker run -d \
   -p 9000:9000 \
-  -e OPENAI_API_KEY=sk-... \
+  -e LLM_API_KEY=sk-... \
   -e MCP_CONFIG=/app/mcp_config.json \
   knucklessg1/github-agent:latest
 ```
@@ -211,7 +211,7 @@ services:
     environment:
       - PROVIDER=openai
       - MODEL_ID=gpt-4o
-      - OPENAI_API_KEY=${OPENAI_API_KEY}
+      - LLM_API_KEY=${LLM_API_KEY}
     volumes:
       - ./mcp_config.json:/app/mcp_config.json
 ```

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "github-agent"
-version = "0.2.0"
+version = "0.2.2"
 readme = "README.md"
 description = "GitHub Agent for MCP"
 requires-python = ">=3.10"