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

王铁柱子哟-

已于 2025-02-12 15:45:05 修改

阅读量286

点赞数 3

分类专栏：杂七杂八的学习笔记文章标签： python llama

于 2025-02-12 15:44:13 首次发布

本文链接：https://blog.csdn.net/wtzszzx/article/details/145593724

版权

杂七杂八的学习笔记专栏收录该内容

5 篇文章

订阅专栏

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

测试代码ollama_API_test.py

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"
)