使用 Python 調用本地 Ollama 的 API 是個很實用的任務，這裏記錄一下實現過程，包括環境準備、集成步驟、配置詳解、實戰應用、排錯指南及性能優化等。

環境準備

在開始之前，我們需要確保我們的開發環境與 Ollama 和 Python 的版本兼容。以下是我整理的版本兼容性矩陣：

接下來，請確保已安裝這些庫和工具。可以通過以下命令安裝所需的 Python 庫：

pip install requests

集成步驟

集成步驟主要包括如何調用Ollama的API接口。以下是基本的接口調用示例。

<details> <summary>多環境適配方案</summary>

在不同的開發環境中，你可能需要根據不同的環境變量配置 API 地址。這是一個簡單的示例：

import os

API_URL = os.getenv("OLLAMA_API_URL", "http://localhost:8000/api")

</details>

以下是一個簡單的請求示例：

import requests

def call_ollama_api(prompt):
    url = "http://localhost:8000/api/generate"
    response = requests.post(url, json={"prompt": prompt})
    
    if response.status_code == 200:
        return response.json()
    else:
        raise Exception(f"Error: {response.status_code}")

配置詳解

我們在 API 調用時需要傳入不同的參數。理解這些參數的含義能夠幫助我們更好地使用 API。以下是一些關鍵參數及其對應關係。

示例代碼片段標記了關鍵參數：

payload = {
    "prompt": prompt,
    "max_tokens": 100,
    "temperature": 0.7
}

實戰應用

下面是一個完整的應用示例，它實現了基本的異常處理機制。我們將通過 GitHub Gist 嵌入這個完整的項目代碼。

import requests

def generate_text(prompt):
    try:
        api_url = "http://localhost:8000/api/generate"
        response = requests.post(api_url, json={"prompt": prompt})
        
        response.raise_for_status()  # 檢查請求是否成功
        
        return response.json().get('text', 'No text returned')

    except requests.exceptions.HTTPError as err:
        print(f"HTTP error occurred: {err}")
    except Exception as e:
        print(f"An error occurred: {e}")

# 示例使用
result = generate_text("Tell me a joke.")
print(result)

引用: 通過調用本地 Ollama 的 API，可以實現高效的文本生成，增強應用的交互性與用户體驗。

排錯指南

在實際使用中，我們可能會遇到一些常見的一系列錯誤。以下是一些可能的錯誤及其排查路徑。

mindmap
  root
    常見錯誤
      網絡連接問題
        檢查本地 server 是否已啓動
      JSON 格式錯誤
        檢查請求體和響應體格式
      404 錯誤
        確保 API 路由正確

以下是一個簡單的代碼修復示例對比：

- response = requests.post(url, json={"prompt": prompt})
+ response = requests.post(url, json={"prompt": prompt}, headers={"Content-Type": "application/json"})

性能優化

進行性能評測時，我們可以用基準測試工具來檢查 API 的性能。以下是我的一些測試結果，顯示了每分鐘的請求處理能力和延遲時間。

這些數據可以幫助我們更好地理解 API 的性能表現，進而做出相應的優化措施。