Files
LLM_tests_and_code_execution/pruebas_ollama.py
T

13 lines
486 B
Python

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)