python+openai兼容的千问api实现连接sse的mcp客户端代码

代码语言:python

所属分类:其他

代码描述:python+openai兼容的千问api实现连接sse的mcp客户端代码,支持chatgpt及兼容openai的国内大模型,大模型api需要支持function call

代码标签: python openai 兼容 千问 api 连接 sse mcp 客户端 代码

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

#!/usr/local/python3/bin/python3
# -*- coding: utf-8 -*
import asyncio
import os
import json
import sys
from typing import Optional
from contextlib import AsyncExitStack
from openai import OpenAI
from dotenv import load_dotenv
from mcp import ClientSession
from mcp.client.sse import sse_client

# 加载 .env 文件,确保 API Key 受到保护
load_dotenv()
API_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"#可以是openai的api
API_KEY = "sk-"#自己申请
MAIN_MODEL = "qwen-plus"#可以是gpt4o

class MCPClient:
    def __init__(self):
        """初始化 MCP 客户端"""
        self.exit_stack = AsyncExitStack()
        self.openai_api_key = API_KEY # 读取 OpenAI API Key
        self.base_url = API_URL # 读取 BASE URL
        self.model =MAIN_MODEL  # 读取模型名称
        # 从环境变量读取配置
        # self.openai_api_key = os.getenv("OPENAI_API_KEY")  # 读取 OpenAI API Key
        # s.........完整代码请登录后点击上方下载按钮下载查看

网友评论0