import getpass
import os

os.environ["NEBULA_API_KEY"]= getpass.getpass()

from langchain_community.chat_models.symblai_nebulaimport ChatNebula
from langchain_core.messagesimport AIMessage, HumanMessage, SystemMessage

chat= ChatNebula(max_tokens=1024, temperature=0.5)

messages=[
    SystemMessage(
        content="You are a helpful assistant that answers general knowledge questions."
),
    HumanMessage(content="What is the capital of France?"),
]
chat.invoke(messages)

AIMessage(content=[{'role': 'human', 'text': 'What is the capital of France?'}, {'role': 'assistant', 'text': 'The capital of France is Paris.'}])

await chat.ainvoke(messages)

AIMessage(content=[{'role': 'human', 'text': 'What is the capital of France?'}, {'role': 'assistant', 'text': 'The capital of France is Paris.'}])

for chunkin chat.stream(messages):
print(chunk.content, end="", flush=True)

 The capital of France is Paris.

chat.batch([messages])

[AIMessage(content=[{'role': 'human', 'text': 'What is the capital of France?'}, {'role': 'assistant', 'text': 'The capital of France is Paris.'}])]

from langchain_core.promptsimport ChatPromptTemplate

prompt= ChatPromptTemplate.from_template("Tell me a joke about {topic}")
chain= prompt| chat

chain.invoke({"topic":"cows"})

AIMessage(content=[{'role': 'human', 'text': 'Tell me a joke about cows'}, {'role': 'assistant', 'text': "Sure, here's a joke about cows:\n\nWhy did the cow cross the road?\n\nTo get to the udder side!"}])