from agensight import init, trace, span
import openai # pip install openai
# Initialize Agensight
init(name="my-first-llm-project")
@trace("joke_generation_workflow")
def generate_a_joke():
@span()
def call_openai_for_joke():
return openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": "Tell me a fun fact about programming"}]
)
print("Calling LLM to get a fun fact...")
response = call_openai_for_joke()
fun_fact = response.choices[0].message.content
print(f"Fun Fact: {fun_fact}")
return fun_fact
if __name__ == "__main__":
generate_a_joke()