wehub-resource-sync
34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
from typing import ClassVar
|
|
|
|
from semantic_kernel.kernel_pydantic import KernelBaseSettings
|
|
|
|
|
|
class OllamaSettings(KernelBaseSettings):
|
|
"""Ollama settings.
|
|
|
|
The settings are first loaded from environment variables with
|
|
the prefix 'OLLAMA_'.
|
|
If the environment variables are not found, the settings can
|
|
be loaded from a .env file with the encoding 'utf-8'.
|
|
If the settings are not found in the .env file, the settings
|
|
are ignored; however, validation will fail alerting that the
|
|
settings are missing.
|
|
|
|
Required settings for prefix 'OLLAMA' are:
|
|
- chat_model_id: str - The chat model ID. (Env var OLLAMA_CHAT_MODEL_ID)
|
|
- text_model_id: str - The text model ID. (Env var OLLAMA_TEXT_MODEL_ID)
|
|
- embedding_model_id: str - The embedding model ID. (Env var OLLAMA_EMBEDDING_MODEL_ID)
|
|
|
|
Optional settings for prefix 'OLLAMA' are:
|
|
- host: HttpsUrl - The endpoint of the Ollama service. (Env var OLLAMA_HOST)
|
|
"""
|
|
|
|
env_prefix: ClassVar[str] = "OLLAMA_"
|
|
|
|
chat_model_id: str | None = None
|
|
text_model_id: str | None = None
|
|
embedding_model_id: str | None = None
|
|
host: str | None = None
|