bhai-cli 0.2.3__tar.gz → 0.2.4__tar.gz

{bhai_cli-0.2.3 → bhai_cli-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bhai-cli
-Version: 0.2.3
+Version: 0.2.4
 Summary: BHAI — The Dual-Brain AI Coding Agent with Punjabi Swagger. Indic-native voice + universal LLM tool-calling.
 License: Apache-2.0
 License-File: LICENSE

{bhai_cli-0.2.3 → bhai_cli-0.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "bhai-cli"
-version = "0.2.3"
+version = "0.2.4"
 description = "BHAI — The Dual-Brain AI Coding Agent with Punjabi Swagger. Indic-native voice + universal LLM tool-calling."
 authors = ["Madhav Kapila <smartatk04@gmail.com>"]
 readme = "README.md"

{bhai_cli-0.2.3 → bhai_cli-0.2.4}/src/bhai_cli/orchestrator.py

@@ -62,7 +62,8 @@ CODER_SYSTEM_PROMPT = (
     "5. Use read_file to see exact file contents before edit_file.\n"
     "6. Use execute_bash for running tests, git, docker, and system commands.\n"
     "7. Verify your changes work by running relevant commands.\n"
-    "8. Report results factually — what you did, what worked, what failed.\n\n"
+    "8. Report results factually — what you did, what worked, what failed.\n"
+    "9. IMPORTANT: When you are done with a task or providing an explanation, simply output plain text. DO NOT format your explanation as a JSON tool call if no tools are needed.\n\n"
     "## Tools Available:\n"
     "- list_directory: Explore project structure\n"
     "- codebase_context: Get function/class signatures (AST-based)\n"
@@ -106,6 +107,8 @@ class BhaiOrchestrator:
             }
             if self.config.vibe.api_base:
                 kwargs["api_base"] = self.config.vibe.api_base
+            if self.config.vibe.api_key:
+                kwargs["api_key"] = self.config.vibe.api_key
             response = await litellm.acompletion(**kwargs)
             title = response.choices[0].message.content.strip().strip('"\'')
             if title:
@@ -163,6 +166,8 @@ class BhaiOrchestrator:
         }
         if self.config.vibe.api_base:
             kwargs["api_base"] = self.config.vibe.api_base
+        if self.config.vibe.api_key:
+            kwargs["api_key"] = self.config.vibe.api_key
 
         try:
             response = await litellm.acompletion(**kwargs)
@@ -232,6 +237,8 @@ class BhaiOrchestrator:
             }
             if self.config.coder.api_base:
                 kwargs["api_base"] = self.config.coder.api_base
+            if self.config.coder.api_key:
+                kwargs["api_key"] = self.config.coder.api_key
 
             try:
                 response = await litellm.acompletion(**kwargs)
@@ -269,6 +276,18 @@ class BhaiOrchestrator:
                     })
 
             except Exception as e:
+                error_str = str(e)
+                if "failed_generation" in error_str:
+                    try:
+                        json_start = error_str.find("{")
+                        if json_start != -1:
+                            error_json = json.loads(error_str[json_start:])
+                            failed_gen = error_json.get("error", {}).get("failed_generation", "")
+                            if failed_gen:
+                                return failed_gen
+                    except Exception:
+                        pass
+
                 error = f"CoderAgent error ({type(e).__name__}): {e}"
                 console.print(f"[bold red]⚠ {error}[/]")
                 return error
@@ -313,6 +332,8 @@ class BhaiOrchestrator:
                 }
                 if self.config.vibe.api_base:
                     kwargs["api_base"] = self.config.vibe.api_base
+                if self.config.vibe.api_key:
+                    kwargs["api_key"] = self.config.vibe.api_key
 
                 response = await litellm.acompletion(**kwargs)
                 summary = (response.choices[0].message.content or "Conversation compressed.").strip()