Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve LLMs and Embedding models resources experience (#21)

Browse Source 
* Fix inconsistent default values
* Disallow LLM's empty name. Handle LLM creation error on UI
This commit is contained in:
Duc Nguyen (john)
2024-04-11 07:50:53 +07:00
committed by GitHub
parent f3e82b2e70
commit 3ed50b0f10
3 changed files with 21 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
libs/kotaemon/kotaemon/embeddings/openai.py
View File

@@ -19,7 +19,7 @@ class BaseOpenAIEmbeddings(BaseEmbeddings):
_dependencies = ["openai"]
api_key: str = Param(help="API key", required=True)
api_key: str = Param(None, help="API key", required=True)
timeout: Optional[float] = Param(None, help="Timeout for the API request.")
max_retries: Optional[int] = Param(
None, help="Maximum number of retries for the API request."
@@ -88,6 +88,7 @@ class OpenAIEmbeddings(BaseOpenAIEmbeddings):
base_url: Optional[str] = Param(None, help="OpenAI base URL")
organization: Optional[str] = Param(None, help="OpenAI organization")
model: str = Param(
None,
help=(
"ID of the model to use. You can go to [Model overview](https://platform."
"openai.com/docs/models/overview) to see the available models."
@@ -131,14 +132,16 @@ class OpenAIEmbeddings(BaseOpenAIEmbeddings):
class AzureOpenAIEmbeddings(BaseOpenAIEmbeddings):
azure_endpoint: str = Param(
None,
help=(
"HTTPS endpoint for the Azure OpenAI model. The azure_endpoint, "
"azure_deployment, and api_version parameters are used to construct "
"the full URL for the Azure OpenAI model."
)
),
required=True,
)
azure_deployment: str = Param(help="Azure deployment name", required=True)
api_version: str = Param(help="Azure model version", required=True)
azure_deployment: str = Param(None, help="Azure deployment name", required=True)
api_version: str = Param(None, help="Azure model version", required=True)
azure_ad_token: Optional[str] = Param(None, help="Azure AD token")
azure_ad_token_provider: Optional[str] = Param(None, help="Azure AD token provider")