diff --git a/autollm/auto/llm.py b/autollm/auto/llm.py index 1c2bcaf1..50a8df8a 100644 --- a/autollm/auto/llm.py +++ b/autollm/auto/llm.py @@ -28,6 +28,7 @@ def from_defaults( https://docs.litellm.ai/docs/providers for a list of supported models. max_tokens: The maximum number of tokens to generate by the LLM. temperature: The temperature to use when sampling from the distribution. + system_prompt: The system prompt to use for the LLM. api_base: The API base URL to use for the LLM. Returns: