m-ahmad-official commited on
Commit
d0b1c4e
·
1 Parent(s): 8aac0dd
Files changed (1) hide show
  1. agent.py +14 -19
agent.py CHANGED
@@ -24,27 +24,22 @@ from openai import AsyncOpenAI
24
  # Load environment first
25
  load_dotenv()
26
 
27
- # Get API key from environment
28
- OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
29
- OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
 
 
 
30
 
31
- if not OPENAI_API_KEY and not OPENROUTER_API_KEY:
32
- raise ValueError(
33
- "Neither OPENAI_API_KEY nor OPENROUTER_API_KEY environment variable is set"
34
- )
 
35
 
36
- # Initialize client based on available keys
37
- if OPENAI_API_KEY:
38
- client = AsyncOpenAI(api_key=OPENAI_API_KEY)
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