import ollama

def stream_llama_responses(prompt):
    stream = ollama.chat(model="llama3.1", messages=[{"role": "user", "content": prompt}], stream=True)
    for chunk in stream:
        if "messages" in chunk:
            print(chunk["messages"][0]["content"], end="")
        else:
            print("\nUnexpected response format:", chunk)

if __name__ == "__main__":
    prompt = "Escribe un artículo sobre marketing digital optimizado para SEO."
    stream_llama_responses(prompt)