diff --git a/docs/my-website/docs/observability/literalai_integration.md b/docs/my-website/docs/observability/literalai_integration.md index 5a643ad47..128c86b2c 100644 --- a/docs/my-website/docs/observability/literalai_integration.md +++ b/docs/my-website/docs/observability/literalai_integration.md @@ -22,6 +22,7 @@ import os os.environ["LITERAL_API_KEY"] = "" os.environ['OPENAI_API_KEY']= "" +os.environ['LITERAL_BATCH_SIZE'] = "1" # You won't see logs appear until the batch is full and sent litellm.success_callback = ["literalai"] # Log Input/Output to LiteralAI litellm.failure_callback = ["literalai"] # Log Errors to LiteralAI @@ -46,6 +47,7 @@ import os os.environ["LITERAL_API_KEY"] = "" os.environ['OPENAI_API_KEY']= "" +os.environ['LITERAL_BATCH_SIZE'] = "1" # You won't see logs appear until the batch is full and sent litellm.input_callback = ["literalai"] # Support other Literal AI decorators and prompt templates litellm.success_callback = ["literalai"] # Log Input/Output to LiteralAI @@ -60,7 +62,8 @@ def my_agent(question: str): model="gpt-3.5-turbo", messages=[ {"role": "user", "content": question} - ] + ], + metadata={"literalai_parent_id": literalai_client.get_current_step().id} ) return response diff --git a/docs/my-website/sidebars.js b/docs/my-website/sidebars.js index 0ab40a1c5..e159418f3 100644 --- a/docs/my-website/sidebars.js +++ b/docs/my-website/sidebars.js @@ -273,6 +273,7 @@ const sidebars = { "observability/langfuse_integration", "observability/gcs_bucket_integration", "observability/langsmith_integration", + "observability/literalai_integration", "observability/opentelemetry_integration", "observability/logfire_integration", "observability/arize_integration",