fix groq client (#155)

This commit is contained in:
Ayush Paul 2025-07-08 11:25:21 -04:00 committed by GitHub
parent d16660b88f
commit aaa0904482
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions

View File

@ -1,12 +1,12 @@
from anthropic import AsyncAnthropic
from google import genai
from groq import Groq
from groq import AsyncGroq
from openai import AsyncOpenAI
from src.config import settings
from src.utils.types import Providers
clients: dict[Providers, AsyncAnthropic | AsyncOpenAI | genai.Client | Groq] = {}
clients: dict[Providers, AsyncAnthropic | AsyncOpenAI | genai.Client | AsyncGroq] = {}
if settings.LLM.ANTHROPIC_API_KEY:
anthropic = AsyncAnthropic(api_key=settings.LLM.ANTHROPIC_API_KEY)
@ -29,5 +29,5 @@ if settings.LLM.GEMINI_API_KEY:
clients["google"] = google
if settings.LLM.GROQ_API_KEY:
groq = Groq(api_key=settings.LLM.GROQ_API_KEY)
groq = AsyncGroq(api_key=settings.LLM.GROQ_API_KEY)
clients["groq"] = groq