chips commited on
Commit
0e18f46
·
1 Parent(s): 80c6e40

gemini added as service in app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -85,6 +85,8 @@ async def forward_request(
85
  ai_vendor = "openai"
86
  elif ai_model in settings.ANTHROPIC_MODELS:
87
  ai_vendor = "anthropic"
 
 
88
  service = AIServiceFactory.get_service(ai_vendor)
89
 
90
  try:
 
85
  ai_vendor = "openai"
86
  elif ai_model in settings.ANTHROPIC_MODELS:
87
  ai_vendor = "anthropic"
88
+ elif ai_model in settings.GEMINI_MODELS:
89
+ ai_vendor = "gemini"
90
  service = AIServiceFactory.get_service(ai_vendor)
91
 
92
  try: