简介
本文介绍如何使用 Ollama API 来自动生成 Python 代码。我们将以生成一个计算数字列表总和的函数为例,展示整个实现过程。
实现步骤
1. 环境准备
首先需要导入必要的Python模块:
import requests # 用于发送HTTP请求
import json # 用于处理JSON数据
import subprocess # 用于执行生成的代码
2. 配置API请求
设置 Ollama API 的访问地址和请求参数:
api_url = "http://localhost:11434/api/generate"
data = {
"model": "codegemma:2b", # 使用的AI模型
"prompt": "编写一个Python函数来计算数字列表的总和。\ndef calculate_sum(numbers):\n# 调用函数并打印结果\n# 示例: print(calculate_sum([1, 2, 3, 4, 5]))",
"options": {
"temperature"