github-agent 0.2.1__tar.gz → 0.2.4__tar.gz

{github_agent-0.2.1 → github_agent-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 0.2.1
+Version: 0.2.4
 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.1*
+*Version: 0.2.4*
 
 ## 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.1 → github_agent-0.2.4}/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.1*
+*Version: 0.2.4*
 
 ## 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.1 → github_agent-0.2.4}/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.1"
+__version__ = "0.2.4"
 
 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"))
@@ -274,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,
@@ -297,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)
@@ -352,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)
@@ -569,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}"
@@ -581,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):
@@ -695,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"
@@ -753,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.1 → github_agent-0.2.4}/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,135 @@ 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,
 ):
+    """
+    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
+    """
+    # Create a custom HTTP client if SSL verification is disabled
+    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("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["OPENAI_BASE_URL"] = target_base_url
         if 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"
+
+        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["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["ANTHROPIC_API_KEY"] = api_key
+
+        # AnthropicModel supports http_client directly via some paths,
+        # but pydantic-ai might prefer we pass the client to the provider or use a custom client
+
+        try:
+            if http_client and AsyncAnthropic and AnthropicProvider:
+                client = AsyncAnthropic(
+                    api_key=api_key or os.environ.get("ANTHROPIC_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":
         if api_key:
             os.environ["GEMINI_API_KEY"] = api_key
-            os.environ["GOOGLE_API_KEY"] = api_key
+        # Google SSL disable is tricky with genai, skipping for now unless specifically requested/researched
         return GoogleModel(model_name=model_id)
 
+    elif provider == "groq":
+        if 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("GROQ_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["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
+            # client = Mistral(...) - Mistral SDK might be different
+            # Skipping Mistral custom client for now to avoid breaking without verification
+            # If user needs Mistral SSL disable, we'll need to research Mistral SDK + Provider
+
+        return MistralModel(model_name=model_id)
+
     elif provider == "huggingface":
         if api_key:
-            os.environ["HF_TOKEN"] = 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.1 → github_agent-0.2.4}/github_agent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 0.2.1
+Version: 0.2.4
 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.1*
+*Version: 0.2.4*
 
 ## 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.4/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.1 → github_agent-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "github-agent"
-version = "0.2.1"
+version = "0.2.4"
 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.1/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