Spaces:
Running
Running
Commit ·
d0b1c4e
1
Parent(s): 8aac0dd
update
Browse files
agent.py
CHANGED
|
@@ -24,27 +24,22 @@ from openai import AsyncOpenAI
|
|
| 24 |
# Load environment first
|
| 25 |
load_dotenv()
|
| 26 |
|
| 27 |
-
#
|
| 28 |
-
|
| 29 |
-
|
|
|
|
|
|
|
|
|
|
| 30 |
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
|
|
|
| 35 |
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
third_party_model = OpenAIChatCompletionsModel(openai_client=client, model="gpt-4")
|
| 40 |
-
elif OPENROUTER_API_KEY:
|
| 41 |
-
client = AsyncOpenAI(
|
| 42 |
-
api_key=OPENROUTER_API_KEY,
|
| 43 |
-
base_url="https://openrouter.ai/api/v1",
|
| 44 |
-
)
|
| 45 |
-
third_party_model = OpenAIChatCompletionsModel(
|
| 46 |
-
openai_client=client, model="openai/gpt-4-turbo-preview"
|
| 47 |
-
)
|
| 48 |
|
| 49 |
|
| 50 |
# Make backend package importable
|
|
|
|
| 24 |
# Load environment first
|
| 25 |
load_dotenv()
|
| 26 |
|
| 27 |
+
# OpenRouter configuration - primary choice
|
| 28 |
+
OPENROUTER_API_KEY = os.getenv(
|
| 29 |
+
"OPENROUTER_API_KEY",
|
| 30 |
+
"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",
|
| 31 |
+
)
|
| 32 |
+
OPENROUTER_MODEL = os.getenv("OPENROUTER_MODEL", "tencent/hy3-preview:free")
|
| 33 |
|
| 34 |
+
# Initialize OpenRouter client
|
| 35 |
+
client = AsyncOpenAI(
|
| 36 |
+
api_key=OPENROUTER_API_KEY,
|
| 37 |
+
base_url="https://openrouter.ai/api/v1",
|
| 38 |
+
)
|
| 39 |
|
| 40 |
+
third_party_model = OpenAIChatCompletionsModel(
|
| 41 |
+
openai_client=client, model=OPENROUTER_MODEL
|
| 42 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
|
| 44 |
|
| 45 |
# Make backend package importable
|