fix + add general env key
This commit is contained in:
1 file changed
+16
-4
+16
-4
@@ -19,9 +19,21 @@ if TYPE_CHECKING:
|
|||||||
)
|
)
|
||||||
DEFAULT_PROMPT_PATH = Path("prompt.md")
|
DEFAULT_PROMPT_PATH = Path("prompt.md")
|
||||||
DEFAULT_SESSION_PATH = Path("session/default.jsonl")
|
DEFAULT_SESSION_PATH = Path("session/default.jsonl")
|
||||||
MODEL_ENV_NAMES = ("CEREBRAS_MODEL_STRING", "GROQ_MODEL_STRING")
|
MODEL_ENV_NAMES = (
|
||||||
BASE_URL_ENV_NAMES = ("CEREBRAS_BASE_URL", "GROQ_BASE_URL")
|
"MODEL_NAME",
|
||||||
API_KEY_ENV_NAMES = ("CEREBRAS_API_KEY", "GROQ_API_KEY")
|
"GROQ_MODEL_STRING",
|
||||||
|
"CEREBRAS_MODEL_STRING",
|
||||||
|
)
|
||||||
|
BASE_URL_ENV_NAMES = (
|
||||||
|
"MODEL_BASE_URL",
|
||||||
|
"GROQ_BASE_URL",
|
||||||
|
"CEREBRAS_BASE_URL",
|
||||||
|
)
|
||||||
|
API_KEY_ENV_NAMES = (
|
||||||
|
"MODEL_API_KEY",
|
||||||
|
"GROQ_API_KEY",
|
||||||
|
"CEREBRAS_API_KEY",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def utc_now_iso() -> str:
|
def utc_now_iso() -> str:
|
||||||
@@ -187,7 +199,7 @@ def messages_from_session(
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
return cast(list[ChatCompletionMessageParam], messages)
|
return cast("list[ChatCompletionMessageParam]", messages)
|
||||||
|
|
||||||
|
|
||||||
def tool_result_content(payload: dict[str, Any]) -> str:
|
def tool_result_content(payload: dict[str, Any]) -> str:
|
||||||
|
|||||||
Reference in new issue
Block a user