mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-07-13 11:22:14 +02:00
support for cerebras
This commit is contained in:
parent
97777e9ccf
commit
3477714554
3 changed files with 56 additions and 0 deletions
|
|
@ -672,6 +672,11 @@ def create_llm_service_from_provider(
|
|||
model=model,
|
||||
temperature=temperature if temperature is not None else 0.5,
|
||||
),
|
||||
elif provider == ServiceProviders.CEREBRAS.value:
|
||||
from pipecat.services.cerebras.llm import CerebrasLLMService, CerebrasLLMSettings
|
||||
return CerebrasLLMService(
|
||||
api_key=api_key,
|
||||
settings=CerebrasLLMSettings(model=model, temperature=0.1),
|
||||
)
|
||||
else:
|
||||
raise HTTPException(status_code=400, detail=f"Invalid LLM provider {provider}")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue