dataflow.importers.litellm_importer

Fixed LiteLLM Importer Module.

This module imports LLM and embedding models from LiteLLM data and other sources into Supabase, properly handling all models without limits.

Functions

add_import_log(entity_name, entity_type, status, message)	Add an import log entry to the audit.import_logs table.
get_or_create_provider(provider_name, provider_type)	Get or create a provider and return its data.
get_or_create_provider_type(type_name, display_name)	Get or create a provider type and return its ID.
import_embedding_models()	Import embedding models.
import_from_env()	Extract embedding models from environment variables.
import_llm_models()	Import LLM models from LiteLLM.
main()	Main function to run the import.

Module Contents

dataflow.importers.litellm_importer.add_import_log(entity_name, entity_type, status, message)

Add an import log entry to the audit.import_logs table.

Parameters:

• entity_name (str)

• entity_type (str)

• status (str)

• message (str)

Return type:

None

dataflow.importers.litellm_importer.get_or_create_provider(provider_name, provider_type)

Get or create a provider and return its data.

Parameters:

• provider_name (str)

• provider_type (str)

Return type:

dict[str, Any] | None

dataflow.importers.litellm_importer.get_or_create_provider_type(type_name, display_name)

Get or create a provider type and return its ID.

Parameters:

• type_name (str)

• display_name (str)

Return type:

str | None

dataflow.importers.litellm_importer.import_embedding_models()

Import embedding models.

Returns the number of models imported.

Return type:

int

dataflow.importers.litellm_importer.import_from_env()

Extract embedding models from environment variables.

Look for vars like OPENAI_EMBEDDING_MODEL, AZURE_EMBEDDING_MODEL, etc.

Return type:

list[dict[str, Any]]

dataflow.importers.litellm_importer.import_llm_models()

Import LLM models from LiteLLM.

Returns the number of models imported.

Return type:

int

dataflow.importers.litellm_importer.main()

Main function to run the import.