python调用agnesai的api实现文本聊天、文生图、图片编辑、文生视频、图生视频、多图参考生图或视频示例代码-python代码-BFW代码库

python调用agnesai的api实现文本聊天、文生图、图片编辑、文生视频、图生视频、多图参考生图或视频示例代码

代码语言：python

所属分类：其他

代码描述：python调用agnesai的api实现文本聊天、文生图、图片编辑、文生视频、图生视频、多图参考生图或视频示例代码

代码标签： python 调用agnes ai api 文本聊天文生图图片编辑文生视频图生视频多图参

下面为部分代码预览，完整代码请点击下载或在bfwstudio webide中打开

#apikey申请地址;https://platform.agnes-ai.com/settings/apiKeys
import os
import time
import requests
from typing import List, Optional, Dict, Any

# 请从环境变量设置您的 API Key，或者直接替换为实际 key（不推荐硬编码）
API_KEY = os.environ.get("AGNES_API_KEY", "YOUR_API_KEY")
BASE_URL = "https://apihub.agnes-ai.com"


def chat_completion(
    messages: List[Dict[str, str]],
    model: str = "agnes-2.0-flash",
    stream: bool = False,
    **kwargs
) -> Dict[str, Any]:
    """
    文本聊天补全
    :param messages: 消息列表，例如 [{"role": "user", "content": "Hello"}]
    :param model: 模型名称
    :param stream: 是否流式输出
    :param kwargs: 其他参数（如 temperature, max_tokens 等）
    :return: API 响应（流式时返回生成器，非流式返回完整响应字典）
    """
    url = f"{BASE_URL}/v1/chat/completions"
    headers = {
        "Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }
    payload = {
        "model": model,
        "messages": messages,
        "stream": stream,
        **kwargs
    }
    if stream:
        # 流式请求，返回响应迭代器
        response = requests.post(url, headers=headers, json=payload, stream=True)
        response.raise_for_status()
        return response  # 调用方可以迭代 response.iter_lines()
    else:
        response = requests.post(url, headers=headers, json=payload)
        response.raise_for_status()
        return response.json()


def generate_image(
    prompt: str,
    model: str = "agnes-im.........完整代码请登录后点击上方下载按钮下载查看