OpenLLM

pydantic model llama_index.llms.openllm.OpenLLM

OpenLLM LLM.

Show JSON schema

{
   "title": "OpenLLM",
   "description": "OpenLLM LLM.",
   "type": "object",
   "properties": {
      "callback_manager": {
         "title": "Callback Manager"
      },
      "model_id": {
         "title": "Model Id",
         "description": "Given Model ID from HuggingFace Hub. This can be either a pretrained ID or local path. This is synonymous to HuggingFace's '.from_pretrained' first argument",
         "type": "string"
      },
      "model_version": {
         "title": "Model Version",
         "description": "Optional model version to save the model as.",
         "type": "string"
      },
      "model_tag": {
         "title": "Model Tag",
         "description": "Optional tag to save to BentoML store.",
         "type": "string"
      },
      "prompt_template": {
         "title": "Prompt Template",
         "description": "Optional prompt template to pass for this LLM.",
         "type": "string"
      },
      "system_message": {
         "title": "System Message",
         "description": "Optional system message to pass for this LLM.",
         "type": "string"
      },
      "backend": {
         "title": "Backend",
         "description": "Optional backend to pass for this LLM. By default, it will use vLLM if vLLM is available in local system. Otherwise, it will fallback to PyTorch.",
         "enum": [
            "vllm",
            "pt"
         ],
         "type": "string"
      },
      "quantize": {
         "title": "Quantize",
         "description": "Optional quantization methods to use with this LLM. See OpenLLM's --quantize options from `openllm start` for more information.",
         "enum": [
            "awq",
            "gptq",
            "int8",
            "int4",
            "squeezellm"
         ],
         "type": "string"
      },
      "serialization": {
         "title": "Serialization",
         "description": "Optional serialization methods for this LLM to be save as. Default to 'safetensors', but will fallback to PyTorch pickle `.bin` on some models.",
         "enum": [
            "safetensors",
            "legacy"
         ],
         "type": "string"
      },
      "trust_remote_code": {
         "title": "Trust Remote Code",
         "description": "Optional flag to trust remote code. This is synonymous to Transformers' `trust_remote_code`. Default to False.",
         "type": "boolean"
      },
      "class_name": {
         "title": "Class Name",
         "type": "string",
         "default": "OpenLLM"
      }
   },
   "required": [
      "model_id",
      "serialization",
      "trust_remote_code"
   ]
}

Config

arbitrary_types_allowed: bool = True

Fields

backend (Optional[Literal['vllm', 'pt']])
model_id (str)
model_tag (Optional[str])
model_version (Optional[str])
prompt_template (Optional[str])
quantize (Optional[Literal['awq', 'gptq', 'int8', 'int4', 'squeezellm']])
serialization (Literal['safetensors', 'legacy'])
system_message (Optional[str])
trust_remote_code (bool)

Validators

_validate_callback_manager » callback_manager

field backend: Optional[Literal['vllm', 'pt']] = None: Optional backend to pass for this LLM. By default, it will use vLLM if vLLM is available in local system. Otherwise, it will fallback to PyTorch.

field model_id: str [Required]: Given Model ID from HuggingFace Hub. This can be either a pretrained ID or local path. This is synonymous to HuggingFace’s ‘.from_pretrained’ first argument

field model_tag: Optional[str] = None: Optional tag to save to BentoML store.

field model_version: Optional[str] = None: Optional model version to save the model as.

field prompt_template: Optional[str] = None: Optional prompt template to pass for this LLM.

field quantize: Optional[Literal['awq', 'gptq', 'int8', 'int4', 'squeezellm']] = None: Optional quantization methods to use with this LLM. See OpenLLM’s –quantize options from openllm start for more information.

field serialization: Literal['safetensors', 'legacy'] [Required]: Optional serialization methods for this LLM to be save as. Default to ‘safetensors’, but will fallback to PyTorch pickle .bin on some models.

field system_message: Optional[str] = None: Optional system message to pass for this LLM.

field trust_remote_code: bool [Required]: Optional flag to trust remote code. This is synonymous to Transformers’ trust_remote_code. Default to False.

async achat(messages: Sequence[ChatMessage], **kwargs: Any) → Any: Async chat endpoint for LLM.

async acomplete(*args: Any, **kwargs: Any) → Any: Async completion endpoint for LLM.

astream_chat(messages: Sequence[ChatMessage], **kwargs: Any) → Any: Async streaming chat endpoint for LLM.

astream_complete(*args: Any, **kwargs: Any) → Any: Async streaming completion endpoint for LLM.

chat(messages: Sequence[ChatMessage], **kwargs: Any) → Any: Chat endpoint for LLM.

classmethod class_name() → str

Get the class name, used as a unique ID in serialization.

This provides a key that makes serialization robust against actual class name changes.

complete(*args: Any, **kwargs: Any) → Any: Completion endpoint for LLM.

stream_chat(messages: Sequence[ChatMessage], **kwargs: Any) → Any: Streaming chat endpoint for LLM.

stream_complete(*args: Any, **kwargs: Any) → Any: Streaming completion endpoint for LLM.

property metadata: LLMMetadata: LLM metadata.

pydantic model llama_index.llms.openllm.OpenLLMAPI

OpenLLM Client interface. This is useful when interacting with a remote OpenLLM server.