聊天机器人企业应用侧重版

import openai
import logging
from typing import Optional

# 配置日志（企业级应用必备：便于问题排查）
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)

class EnterpriseChatbot:
    def __init__(self, api_key: str, base_url: str = 'https://yunwu.ai/v1'):
        """初始化企业级聊天机器人客户端"""
        try:
            self.client = openai.OpenAI(
                base_url=base_url,
                api_key=api_key
            )
            logger.info("OpenAI客户端初始化成功")
        except Exception as e:
            logger.error(f"客户端初始化失败：{str(e)}", exc_info=True)
            raise  # 抛出异常，确保服务启动前发现问题

    def get_chat_response(self, prompt: str, max_tokens: int = 150, engine: str = "davinci") -> Optional[str]:
        """
        企业级聊天响应方法（含异常处理、日志记录）
        :param prompt: 用户输入
        :param max_tokens: 回复最大长度
        :param engine: 调用模型
        :return: 机器人回复（异常时返回None）
        """
        try:
            response = self.client.Completion.create(
                engine=engine,
                prompt=prompt,
                max_tokens=max_tokens,
                temperature=0.7  # 企业级调参：平衡回复准确性与多样性
            )
            reply = response.choices[0].text.strip()
            logger.info(f"生成回复成功，用户输入：{prompt[:20]}...，回复：{reply[:20]}...")
            return reply
        except Exception as e:
            logger.error(f"生成回复失败，用户输入：{prompt[:20]}...，错误信息：{str(e)}", exc_info=True)
            return None

# 企业级应用示例（对接业务系统）
if __name__ == "__main__":
    # 从环境变量获取API密钥（企业级安全实践：避免硬编码）
    import os
    api_key = os.getenv("OPENAI_API_KEY", "your-api-key")
    
    # 初始化机器人
    bot = EnterpriseChatbot(api_key=api_key)
    
    # 模拟客服场景对话
    user_queries = [
        "我的订单还没发货，怎么查物流？",
        "这个产品支持7天无理由退货吗？"
    ]
    
    for query in user_queries:
        reply = bot.get_chat_response(prompt=query)
        print(f"用户：{query}\n机器人：{reply}\n")