python测试ollama的API接口是否可用

python测试ollama的API接口是否可用

测试代码ollama_API_test.py

# ollama serve
# nohup ollama serve > ollama.log 2>&1 &
# ollama list
# ollama run llama3
# Use Ctrl + d or /bye to exit.

import ollama

modele_name = "llama3"
message = """
Why is the sky blue?
"""
response = ollama.chat(
    model=modele_name,
    messages=[{"role": "user", "content": message}],
    stream=False,  # 是否流式输出
)
print(
    "\n"
    + "-" * 30
    + "\n"
    + str(response["message"]["content"].strip("\n").strip())
    + "\n"
    + "-" * 30
    + "\n"
)