bodo.pandas.BodoSeries.ai.embed¶
BodoSeries.ai.embed(
api_key: str,
model: str | None = None,
base_url: str | None = None,
**embed_kwargs) -> BodoSeries
Parameters
- api_key: str: The API key for authentication with the LLM endpoint
- model: str | None: The model to use for embedding. If None is provided, the default model from the endpoint will be used.
- base_url: str: The URL of the OpenAI compatible LLM endpoint.
- **embed_kwargs: dict: Additional keyword arguments for the LLM embedding API.
Returns
- BodoSeries: A series containing the embedded vectors as lists of doubles.
Example
import bodo.pandas as pd
# Example series
a = pd.Series(["bodo.ai will improve your workflows.", "This is a professional sentence."])
# Define the LLM base_url and API key
base_url = "https://api.example.com/v1"
api_key = "your_api_key_here"
# Embed the series using the LLM
b = a.ai.embed(api_key=api_key, model="text-embedding-3-small", base_url=base_url)
print(b)
Output: