Skip to content

Commit f96932e

Browse files
author
SentienceDEV
committed
deep infra models
1 parent caa5131 commit f96932e

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

sentience/llm_provider.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -319,6 +319,26 @@ def model_name(self) -> str:
319319
return self._model_name
320320

321321

322+
class DeepInfraProvider(OpenAIProvider):
323+
"""
324+
DeepInfra provider via OpenAI-compatible API.
325+
326+
Uses DeepInfra's OpenAI-compatible endpoint:
327+
https://api.deepinfra.com/v1/openai
328+
329+
API token is read from DEEPINFRA_TOKEN if not provided.
330+
"""
331+
332+
def __init__(
333+
self,
334+
api_key: str | None = None,
335+
model: str = "meta-llama/Meta-Llama-3-8B-Instruct",
336+
base_url: str = "https://api.deepinfra.com/v1/openai",
337+
):
338+
api_key = get_api_key_from_env(["DEEPINFRA_TOKEN"], api_key)
339+
super().__init__(api_key=api_key, model=model, base_url=base_url)
340+
341+
322342
class AnthropicProvider(LLMProvider):
323343
"""
324344
Anthropic provider implementation (Claude 3 Opus, Sonnet, Haiku, etc.)

0 commit comments

Comments
 (0)