SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 13:55:41 +01:00
Code Issues Projects Releases Packages Activity

FEAT: Added Fireworks Integration

Browse source
This commit is contained in:
Somashekar B R 2024-09-20 14:13:09 +05:30
parent 77461b96cf
commit f1ad995124
7 changed files with 112 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

16
private_gpt/settings/settings.py
View file

@ -260,6 +260,21 @@ class OpenAISettings(BaseModel):
description="OpenAI embedding Model to use. Example: 'text-embedding-3-large'.",
)
class FireWorksSettings(BaseModel):
api_key: str
model: str = Field(
"accounts/fireworks/models/llama-v3p1-70b-instruct",
description="FireWorks Model to use. Example: 'accounts/fireworks/models/llama-v3p1-70b-instruct'.",
)
# embedding_api_base: str = Field(
# None,
# description="Base URL of OpenAI API. Example: 'https://api.openai.com/v1'.",
# )
embedding_api_key: str
# embedding_model: str = Field(
# "text-embedding-ada-002",
# description="OpenAI embedding Model to use. Example: 'text-embedding-3-large'.",
# )
class GeminiSettings(BaseModel):
api_key: str
@ -586,6 +601,7 @@ class Settings(BaseModel):
huggingface: HuggingFaceSettings
sagemaker: SagemakerSettings
openai: OpenAISettings
fireworks: FireWorksSettings
gemini: GeminiSettings
ollama: OllamaSettings
azopenai: AzureOpenAISettings