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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 0.2.4
+Version: 0.2.6
 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.4*
+*Version: 0.2.6*
 
 ## Overview
 

{github_agent-0.2.4 → github_agent-0.2.6}/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.4*
+*Version: 0.2.6*
 
 ## Overview
 

{github_agent-0.2.4 → github_agent-0.2.6}/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.4"
+__version__ = "0.2.6"
 
 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)
 
@@ -726,6 +723,12 @@ def agent_server():
         default=DEFAULT_ENABLE_WEB_UI,
         help="Enable Pydantic AI Web UI",
     )
+
+    parser.add_argument(
+        "--insecure",
+        action="store_true",
+        help="Disable SSL verification for LLM requests (Use with caution)",
+    )
     parser.add_argument("--help", action="store_true", help="Show usage")
 
     args = parser.parse_args()

{github_agent-0.2.4 → github_agent-0.2.6}/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,8 +247,8 @@ 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,
 ):
     """
@@ -271,7 +264,6 @@ def create_model(
     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)
@@ -280,7 +272,6 @@ def create_model(
         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"),
@@ -290,7 +281,6 @@ def create_model(
             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:
@@ -298,7 +288,6 @@ def create_model(
         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"
 
@@ -319,9 +308,6 @@ def create_model(
         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(
@@ -338,7 +324,6 @@ def create_model(
     elif provider == "google":
         if api_key:
             os.environ["GEMINI_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":
@@ -360,12 +345,7 @@ def create_model(
             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)
 

{github_agent-0.2.4 → github_agent-0.2.6}/github_agent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 0.2.4
+Version: 0.2.6
 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.4*
+*Version: 0.2.6*
 
 ## Overview
 

{github_agent-0.2.4 → github_agent-0.2.6}/pyproject.toml

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