From 69f55ea68f2ff2abc2d4b47d847ff24e6a0631fa Mon Sep 17 00:00:00 2001
From: Aditya Singh <adity.347317@gmail.com>
Date: Tue, 2 Dec 2025 14:39:31 +0530
Subject: [PATCH] Minor changes

---
 pr_agent/algo/__init__.py                | 2 +-
 pr_agent/settings/.secrets_template.toml | 5 ++++-
 pr_agent/settings/configuration.toml     | 5 ++++-
 setup-pr-agent.sh                        | 4 +++-
 4 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
index fff33c1a..53e4cd41 100644
--- a/pr_agent/algo/__init__.py
+++ b/pr_agent/algo/__init__.py
@@ -173,7 +173,7 @@ MAX_TOKENS = {
     "mistral/codestral-mamba-latest": 256000,
     "codestral/codestral-latest": 8191,
     "codestral/codestral-2405": 8191,
-    "blackboxai/openai/gpt-4.1": 120000,
+    "openai/blackboxai/openai/gpt-4.1": 120000,
 }
 
 USER_MESSAGE_ONLY_MODELS = [
diff --git a/pr_agent/settings/.secrets_template.toml b/pr_agent/settings/.secrets_template.toml
index 70238c2d..fe2ebe29 100644
--- a/pr_agent/settings/.secrets_template.toml
+++ b/pr_agent/settings/.secrets_template.toml
@@ -7,7 +7,10 @@
 # See README for details about GitHub App deployment.
 
 [openai]
-key = ""  # Acquire through https://platform.openai.com
+# Your OpenAI-compatible API key. For Blackbox.ai, put your Blackbox key here.
+key = "<BLACKBOX_API_KEY>"  # Acquire from your Blackbox.ai account
+# Optional: specify a custom base URL (useful for Blackbox.ai or self-hosted OpenAI-compatible endpoints)
+api_base = "https://api.blackbox.ai/"
 #org = "<ORGANIZATION>"  # Optional, may be commented out.
 # Uncomment the following for Azure OpenAI
 #api_type = "azure"
diff --git a/pr_agent/settings/configuration.toml b/pr_agent/settings/configuration.toml
index 7a6c1a80..fe2a9729 100644
--- a/pr_agent/settings/configuration.toml
+++ b/pr_agent/settings/configuration.toml
@@ -6,7 +6,10 @@
 
 [config]
 # models
-model="gpt-5-2025-08-07"
+# Default model: Blackbox AI with OpenAI-compatible API
+# Provider prefix 'openai/' tells LiteLLM which handler to use;
+# the rest 'blackboxai/openai/gpt-4.1' is passed to the Blackbox API endpoint
+model="openai/blackboxai/openai/gpt-4.1"
 fallback_models=[]
 #model_reasoning="o4-mini" # dedicated reasoning model for self-reflection
 #model_weak="gpt-4o" # optional, a weaker model to use for some easier tasks
diff --git a/setup-pr-agent.sh b/setup-pr-agent.sh
index f19836db..802fce0b 100755
--- a/setup-pr-agent.sh
+++ b/setup-pr-agent.sh
@@ -42,9 +42,11 @@ cat <<EOF > pr_agent.toml
 [general]
 log_level = "info"
 
+# Keep provider as OpenAI-compatible but default to Blackbox model identifier
+# Use Blackbox model with OpenAI provider routing
 [llm]
 provider = "openai"
-model = "gpt-4o"
+model = "openai/blackboxai/openai/gpt-4.1"
 # Add your API key here or via environment variable
 api_key = ""