lmnr

import os import openai from lmnr import Laminar, observe # It's recommended to install with pip install -U 'lmnr[all]' openai python-dotenv # Initialize Laminar. project_api_key will be read from LMNR_PROJECT_API_KEY environment # variable if not explicitly passed. Laminar.initialize( project_api_key=os.environ.get('LMNR_PROJECT_API_KEY', ''), # Instrument the OpenAI module for automatic tracing of API calls. instrumentModules={ "OpenAI": openai } ) # Initialize OpenAI client *after* Laminar has instrumented the module. # The OpenAI API key will be read from OPENAI_API_KEY environment variable. client = openai.OpenAI(api_key=os.environ.get('OPENAI_API_KEY', '')) @observe(name="poem_generator") def generate_poem(topic: str) -> str: """Generates a short poem using OpenAI and traces the call with Laminar.""" print(f"Calling OpenAI to generate a poem about: {topic}") response = client.chat.completions.create( model="gpt-4o-mini", messages=[ {"role": "user", "content": f"Write a short poem about {topic}"}, ], ) return response.choices[0].message.content if __name__ == "__main__": # Ensure LMNR_PROJECT_API_KEY and OPENAI_API_KEY are set in your environment # or a .env file loaded with python-dotenv. if not os.environ.get('LMNR_PROJECT_API_KEY') or not os.environ.get('OPENAI_API_KEY'): print("Please set LMNR_PROJECT_API_KEY and OPENAI_API_KEY environment variables.") print("You can get LMNR_PROJECT_API_KEY from your Laminar dashboard.") exit(1) print("Generating a poem with Laminar tracing enabled...") try: poem = generate_poem("a starry night") print("\nGenerated Poem:") print(poem) print("\nCheck your Laminar dashboard for the trace of this operation!") except Exception as e: print(f"An error occurred: {e}") print("Ensure your API keys are correct and you have network connectivity.")