Przeglądaj źródła

Update chat_service: add openrouter gemini

StrayWarrior 2 dni temu
rodzic
commit
09a8d9ad6c
1 zmienionych plików z 4 dodań i 1 usunięć
  1. 4 1
      pqai_agent/chat_service.py

+ 4 - 1
pqai_agent/chat_service.py

@@ -34,9 +34,10 @@ OPENAI_API_TOKEN = 'sk-proj-6LsybsZSinbMIUzqttDt8LxmNbi-i6lEq-AUMzBhCr3jS8sme9AG
 OPENAI_BASE_URL = 'https://api.openai.com/v1'
 OPENAI_MODEL_GPT_4o = 'gpt-4o'
 OPENAI_MODEL_GPT_4o_mini = 'gpt-4o-mini'
-OPENROUTER_API_TOKEN = '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'
+OPENROUTER_API_TOKEN = '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'
 OPENROUTER_BASE_URL = 'https://openrouter.ai/api/v1/'
 OPENROUTER_MODEL_CLAUDE_3_7_SONNET = 'anthropic/claude-3.7-sonnet'
+OPENROUTER_MODEL_GEMINI_2_5_PRO = 'google/gemini-2.5-pro'
 ALIYUN_API_TOKEN = 'sk-47381479425f4485af7673d3d2fd92b6'
 ALIYUN_BASE_URL = 'https://dashscope.aliyuncs.com/compatible-mode/v1'
 
@@ -92,6 +93,7 @@ class OpenAICompatible:
     ]
     openrouter_models = [
         OPENROUTER_MODEL_CLAUDE_3_7_SONNET,
+        OPENROUTER_MODEL_GEMINI_2_5_PRO
     ]
 
     model_prices = {
@@ -103,6 +105,7 @@ class OpenAICompatible:
         OPENAI_MODEL_GPT_4o: ModelPrice(input_price=2.5, output_price=10, currency='USD'),
         OPENAI_MODEL_GPT_4o_mini: ModelPrice(input_price=0.15, output_price=0.6, currency='USD'),
         OPENROUTER_MODEL_CLAUDE_3_7_SONNET: ModelPrice(input_price=3, output_price=15, currency='USD'),
+        OPENROUTER_MODEL_GEMINI_2_5_PRO: ModelPrice(input_price=1.25, output_price=10, currency='USD'),
     }
 
     @staticmethod