mycode-aiagent 0.1.4__tar.gz → 0.2.1__tar.gz

{mycode_aiagent-0.1.4 → mycode_aiagent-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mycode-aiagent
-Version: 0.1.4
+Version: 0.2.1
 Summary: Style-aware code generation — analyze any codebase and generate new code that matches its style
 License: MIT
 Project-URL: Homepage, https://github.com/RyanAbbottData/MyCode

{mycode_aiagent-0.1.4 → mycode_aiagent-0.2.1}/my_code/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.1.0"
+__version__ = "0.1.5"
 
 from .analyzer import StyleAnalyzer
 from .generator import generate_code

{mycode_aiagent-0.1.4 → mycode_aiagent-0.2.1}/my_code/analyzer.py

@@ -3,7 +3,7 @@ import re
 from pathlib import Path
 
 from .backends import AIBackend
-from .utils.prompts import STYLE_EXTRACTION_PROMPT, STYLE_SUMMARY_PROMPT
+from .utils.prompts import STYLE_EXTRACTION_PROMPT, STYLE_MERGE_PROMPT
 
 _SKIP_EXACT = {"__pycache__", ".git", "node_modules", "dist", "build"}
 
@@ -52,30 +52,32 @@ class StyleAnalyzer:
         if not files:
             raise FileNotFoundError(f"No Python files found under {root}")
 
-        observations = []
+        profile = {}
         for path in files:
             rel = path.relative_to(root)
             if verbose:
                 print(f"  Analyzing {rel} ...")
             obs = self.analyze_file(path)
-            if obs:
-                observations.append(obs)
-
-        if not observations:
+            if not obs:
+                continue
+            if not profile:
+                profile = obs
+                continue
+            prompt = STYLE_MERGE_PROMPT.format(
+                profile=json.dumps(profile, indent=2),
+                observation=json.dumps(obs, indent=2),
+                filename=path.name,
+            )
+            raw = self.backend.ask_to_analyze(prompt)
+            try:
+                profile = _extract_json(raw)
+            except (ValueError, json.JSONDecodeError) as e:
+                print(f"  [warn] Could not merge style from {path.name}: {e}")
+
+        if not profile:
             raise RuntimeError("No style data could be extracted from any file.")
 
-        if len(observations) == 1:
-            return observations[0]
-
-        prompt = STYLE_SUMMARY_PROMPT.format(
-            observations=json.dumps(observations, indent=2)
-        )
-        raw = self.backend.ask_to_analyze(prompt)
-        try:
-            return _extract_json(raw)
-        except (ValueError, json.JSONDecodeError) as e:
-            print(f"[warn] Could not synthesize profile, using first observation: {e}")
-            return observations[0]
+        return profile
 
     @staticmethod
     def save_profile(profile: dict, path: Path):

{mycode_aiagent-0.1.4 → mycode_aiagent-0.2.1}/my_code/backends/__init__.py

@@ -15,6 +15,7 @@ def make_backend(
     ricky_url: str = "http://localhost:8000/mcp",
     mcp_url: str = "http://localhost:8001/mcp",
     model: str | None = None,
+    timeout: int = 120,
 ) -> AIBackend:
     if backend == "claude":
         key = api_key or os.environ.get("ANTHROPIC_API_KEY")
@@ -27,7 +28,7 @@ def make_backend(
             raise ValueError("OpenAI backend requires OPENAI_API_KEY or --api-key")
         return OpenAIBackend(api_key=key, **{"model": model} if model else {})
     if backend == "mcp":
-        return MCPBackend(url=mcp_url)
+        return MCPBackend(url=mcp_url, timeout=timeout)
     if backend == "llama":
-        return RickyBackend(url=ricky_url)
+        return RickyBackend(url=ricky_url, timeout=timeout)
     raise ValueError(f"Unknown backend {backend!r}. Choose: llama, claude, openai, mcp")

{mycode_aiagent-0.1.4 → mycode_aiagent-0.2.1}/my_code/backends/mcp_backend.py

@@ -3,8 +3,8 @@ from ..mcp_client import MCPClient
 
 
 class MCPBackend(AIBackend):
-    def __init__(self, url: str = "http://localhost:8001/mcp"):
-        self._client = MCPClient(url=url)
+    def __init__(self, url: str = "http://localhost:8001/mcp", timeout: int = 120):
+        self._client = MCPClient(url=url, timeout=timeout)
 
     def ask_for_code(self, prompt: str) -> str:
         return self._client.ask_for_code(prompt)

{mycode_aiagent-0.1.4 → mycode_aiagent-0.2.1}/my_code/backends/ricky_backend.py

@@ -5,8 +5,8 @@ from ..mcp_client import MCPClient
 class RickyBackend(AIBackend):
     max_file_chars: int = 1500
 
-    def __init__(self, url: str = "http://localhost:8000/mcp"):
-        self._client = MCPClient(url=url)
+    def __init__(self, url: str = "http://localhost:8000/mcp", timeout: int = 120):
+        self._client = MCPClient(url=url, timeout=timeout)
 
     def ask_for_code(self, prompt: str) -> str:
         return self._client.ask_for_code(prompt)

{mycode_aiagent-0.1.4 → mycode_aiagent-0.2.1}/my_code/cli.py

@@ -29,6 +29,7 @@ def cmd_analyze(args):
         ricky_url=args.ricky_url,
         mcp_url=args.mcp_url,
         model=args.model,
+        timeout=args.timeout,
     )
 
     print(f"Analyzing codebase at {root} ...")
@@ -53,6 +54,7 @@ def cmd_generate(args):
         ricky_url=args.ricky_url,
         mcp_url=args.mcp_url,
         model=args.model,
+        timeout=args.timeout,
     )
 
     profile = json.loads(profile_path.read_text(encoding="utf-8"))
@@ -72,6 +74,7 @@ def main():
     parser.add_argument("--model", default=None, help="Override default model for claude/openai backends")
     parser.add_argument("--ricky-url", default="http://localhost:8000/mcp", help="Ricky MCP server URL (llama backend)")
     parser.add_argument("--mcp-url", default="http://localhost:8001/mcp", help="MCP server URL (mcp backend)")
+    parser.add_argument("--timeout", type=int, default=120, help="Request timeout in seconds (default: 120)")
     parser.add_argument("--profile", default=str(DEFAULT_PROFILE))
 
     sub = parser.add_subparsers(dest="command", required=True)

{mycode_aiagent-0.1.4 → mycode_aiagent-0.2.1}/my_code/mcp_client.py

@@ -24,8 +24,9 @@ class MCPClient:
       3. POST /mcp  tools/call  → invoke a tool
     """
 
-    def __init__(self, url: str = "http://localhost:8001/mcp"):
+    def __init__(self, url: str = "http://localhost:8001/mcp", timeout: int = 120):
         self.url = url
+        self._timeout = timeout
         self._session_id: str | None = None
         self._tool_name: str | None = None
         self._tool_name_analyze: str | None = None
@@ -35,7 +36,7 @@ class MCPClient:
         self._discover_tools()
         print(f"Connected to MCP server — tools: '{self._tool_name}', '{self._tool_name_analyze}'")
 
-    def _post(self, method: str, params: dict, timeout: int = 120) -> dict:
+    def _post(self, method: str, params: dict) -> dict:
         headers = {
             "Content-Type": "application/json",
             "Accept": "application/json, text/event-stream",
@@ -49,7 +50,7 @@ class MCPClient:
             "method": method,
             "params": params,
         }
-        resp = requests.post(self.url, json=payload, headers=headers, timeout=timeout)
+        resp = requests.post(self.url, json=payload, headers=headers, timeout=self._timeout)
         resp.raise_for_status()
 
         if "Mcp-Session-Id" in resp.headers:
@@ -117,5 +118,5 @@ class MCPClient:
         result = self._post("tools/call", {
             "name": self._tool_name_analyze,
             "arguments": {self._analyze_arg: prompt},
-        }, timeout=600)
+        })
         return self._extract_text(result)

{mycode_aiagent-0.1.4 → mycode_aiagent-0.2.1}/my_code/utils/prompts.py

@@ -30,15 +30,20 @@ Source file ({filename}):
 ```
 """
 
-STYLE_SUMMARY_PROMPT = """\
-You are a code style analyst. Below are JSON style observations extracted from multiple files in a codebase. Synthesize them into a single authoritative style profile JSON.
+STYLE_MERGE_PROMPT = """\
+You are a code style analyst. Update the current style profile by merging in a new file observation.
 
-Use the same schema. For fields where files disagree, pick the majority or most consistent value. Add a "confidence" field (low/medium/high) per section. Keep representative_snippets to the 3 best examples across all files.
+Rules:
+- Where they agree, keep the value.
+- Where they disagree, pick the more consistent value.
+- Keep representative_snippets to the 3 best examples total.
+- Return ONLY valid JSON using the same schema.
 
-Return ONLY valid JSON.
+Current profile:
+{profile}
 
-Observations:
-{observations}
+New observation ({filename}):
+{observation}
 """
 
 CODE_GENERATION_PROMPT = """\

{mycode_aiagent-0.1.4 → mycode_aiagent-0.2.1}/mycode_aiagent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mycode-aiagent
-Version: 0.1.4
+Version: 0.2.1
 Summary: Style-aware code generation — analyze any codebase and generate new code that matches its style
 License: MIT
 Project-URL: Homepage, https://github.com/RyanAbbottData/MyCode

{mycode_aiagent-0.1.4 → mycode_aiagent-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mycode-aiagent"
-version = "0.1.4"
+version = "0.2.1"
 description = "Style-aware code generation — analyze any codebase and generate new code that matches its style"
 readme = "README.md"
 requires-python = ">=3.10"