steamship.agents.llms package

Submodules

steamship.agents.llms.openai module

class steamship.agents.llms.openai.OpenAI(client, model_name: str = 'gpt-3.5-turbo', temperature: float = 0.4, *args, generator: PluginInstance)

Bases: LLM

LLM that uses Steamship’s OpenAI plugin to generate completions.

NOTE: By default, this LLM uses the gpt-3.5-turbo model. Valid model choices are gpt-3.5-turbo and gpt-4.

client: Steamship

complete(prompt: str, stop: Optional[str] = None) → List[Block]

Completes the provided prompt, stopping when the stop sequeunce is found.

generator: PluginInstance

Module contents

class steamship.agents.llms.OpenAI(client, model_name: str = 'gpt-3.5-turbo', temperature: float = 0.4, *args, generator: PluginInstance)

Bases: LLM

LLM that uses Steamship’s OpenAI plugin to generate completions.

NOTE: By default, this LLM uses the gpt-3.5-turbo model. Valid model choices are gpt-3.5-turbo and gpt-4.

client: Steamship

complete(prompt: str, stop: Optional[str] = None) → List[Block]

Completes the provided prompt, stopping when the stop sequeunce is found.

generator: PluginInstance