import os, asyncio
from literalai import LiteralClient
from langchain_anthropic import ChatAnthropic
from langchain_core.messages import HumanMessage
from langchain.schema.runnable.config import RunnableConfig
from dotenv import load_dotenv
load_dotenv()
literal_client = LiteralClient(api_key=os.getenv("LITERAL_API_KEY"))
cb = literal_client.langchain_callback()
model = ChatAnthropic(model='claude-3-opus-20240229', anthropic_api_key=os.getenv("ANTHROPIC_API_KEY"))
message = HumanMessage(content="What is the capital of France?")
response = model.invoke([message], config=RunnableConfig(callbacks=[cb]))
literal_client.flush_and_stop()