3 Tir@sZdZddlZddlZddlZddlmZmZmZmZddl Z ddl Z ddl m Z e j e j de jeZGdddZGdd d ZGd d d ZGd d d ZGdddZGdddZddlZddlZddlZddlmZGdddejjZ eddZ!ddZ"d e#dddZ$edkrVddl%Z%e&e%j'dkrPe%j'ddkrPe"ne$dS)!u 简化版AI智能聊天机器人 使用豆包Doubao-Seed-1.6-Flash-250828模型 对接前端网站聊天窗口，智能回复用户问题 N)ListDictOptionalTuple)datetime)levelc@s<eZdZdZd eedddZd eeee edd d Z d S) DoubaoAPIu豆包大模型API调用器doubao-seed-1-6-flash-250828)api_key model_namecCs||_||_d|_dS)Nz9https://ark.cn-beijing.volces.com/api/v3/chat/completions)r r api_url)selfr r rsimple_chatbot.py__init__szDoubaoAPI.__init__ffffff?)messages temperature max_tokensreturnc Csdd|jd}|j|||d}yDddd}tj|j||d|d}|j|j}|d d d d Stk r} ztj d | dSd} ~ XnXdS)u调用豆包API进行对话zapplication/jsonzBearer )z Content-TypeZ Authorization)ZmodelrrZmax_completion_tokensN)httpZhttps)headersjsonZtimeoutproxieschoicesrmessagecontentu豆包API调用失败: u?抱歉，我现在无法回答您的问题，请稍后再试。) r r requestsZpostr Zraise_for_statusr Exceptionloggererror) r rrrrdatarresponseresulterrrchats zDoubaoAPI.chatN)r )rr) __name__ __module__ __qualname____doc__strrrrfloatintr'rrrrrsrc@sNeZdZdZdedddZdeeeeedd d Z eeed d d Z dS) QADatabaseu问答数据库管理器0/root/老业务网站8.30/backend/ai_agent_qa.db)db_pathcCs ||_dS)N)r1)r r1rrrr<szQADatabase.__init__zh)querylanguagelimitrcCs>ytj|j}tj|_|j}g}|jd|||f|j}|rV|jdd|Dt ||kr|jdd|d||t |f|j}|jdd|Dt ||kr&|j |} xl| D]d} t ||krP|jdd| dd| dd| d||t |f|j} |jdd| DqWt |d krd d d d dg} xh| D]`} | |krH|jdd| dd| d||f|j}|jdd|Dt |d krHPqHW|j t }g}x<|D]4}|d|df}||kr|j ||j|qW|d|Stk r8}ztjd|gSd}~XnXdS)u搜索相关问答z[SELECT question, answer, category FROM qa_pairs WHERE question = ? AND language = ? LIMIT ?cSsg|] }t|qSr)dict).0rowrrr Osz(QADatabase.search_qa..z^SELECT question, answer, category FROM qa_pairs WHERE question LIKE ? AND language = ? LIMIT ?%cSsg|] }t|qSr)r7)r8r9rrrr:XszSELECT question, answer, category FROM qa_pairs WHERE (question LIKE ? OR answer LIKE ? OR keywords LIKE ?) AND language = ? LIMIT ?cSsg|] }t|qSr)r7)r8r9rrrr:esru公司u产品u服务u 做什么u业务zqSELECT question, answer, category FROM qa_pairs WHERE (question LIKE ? OR answer LIKE ?) AND language = ? LIMIT ?cSsg|] }t|qSr)r7)r8r9rrrr:rsquestionanswerNu数据库搜索失败: )sqlite3Zconnectr1ZRowZ row_factorycursorZexecuteZfetchallextendlen_extract_keywordsclosesetaddappendr r!r")r r4r5r6Zconnr?ZresultsZ exact_resultsZ fuzzy_resultskeywordskeywordZkeyword_resultsZ core_wordswordZ broad_resultsseenZunique_resultsr%keyr&rrr search_qa?sd      0      zQADatabase.search_qa)textrcs^tjdd|}|j}ddddddd d d d d ddddddddhfdd|D}|ddS)u提取关键词z[^\w\s\u4e00-\u9fff]u的u了u在u是u我u你u他u她u它u们u这u那u什么u怎么u如何u 为什么u吗u呢u吧cs$g|]}|krt|dkr|qS))rA)r8rI) stop_wordsrrr:sz0QADatabase._extract_keywords..Nr3)resubsplit)r rMZwordsrGr)rPrrBs *zQADatabase._extract_keywordsN)r0)r2r3) r(r)r*r+r,rr.rrrLrBrrrrr/9sHr/c@s$eZdZdZeeedddZdS)LanguageDetectoru语言检测器)rMrc Csttjd|}ttjd|}ttjd|}ttjd|j}t|jdd}|dkr`dS||}|jr|t|jnd}||}||} |d krdS|d krd S| d krd S|d krdSdSdS)u检测文本语言z[\u4e00-\u9fff]z \b[a-zA-Z]+\bz[\u0e00-\u0e7f]u[àáạảãâầấậẩẫăằắặẳẵèéẹẻẽêềếệểễìíịỉĩòóọỏõôồốộổỗơờớợởỡùúụủũưừứựửữỳýỵỷỹđ] rNrr2g333333?thg?vig?enN)rArQfindalllowerreplacerS) rMZ chinese_charsZ english_wordsZ thai_charsZvietnamese_charsZ total_charsZ chinese_ratioZ english_ratioZ thai_ratioZvietnamese_ratiorrrdetect_languages(z LanguageDetector.detect_languageN)r(r)r*r+ staticmethodr,r\rrrrrTsrTc @seZdZdZdddddddgd d d d d dgddgddd gdddddddgddddgddgd d!gdd"d#d$d%d&gd'd(d)d*gd+d,gd-d.gdd/d0d1d2d3gd4d5d6d7gd8d9gd:d;gddd?d@gdAdBdCdDgdEdFgdGdHgddIdJdKdLdMdNgdOdPdQdRgdSdTgdUdVgddWZeeeedXdYdZZd[S)\IntentClassifieru意图分类器u你好u您好u嗨u哈喽u 早上好u 下午好u 晚上好ZhellohiZheyz good morningzgood afternoonz good eveninguสวัสดีuหวัดดีu xin chàouchào)r2rXrVrWu公司u企业u你们u介绍u 是什么u 做什么ZcompanyZbusinesszwhat do you doZaboutuบริษัทuธุรกิจucông tyudoanh nghiệpu服务u产品u功能u提供u 解决方案serviceproductZsolutionZofferuบริการuผลิตภัณฑ์u dịch vụu sản phẩmu价格u费用u 多少钱u收费u报价ZpriceZcostZfeezhow muchu ราคาuค่าใช้จ่ายugiáuchi phíu联系u电话u邮箱u微信u地址contactZphoneZemailZaddressuติดต่อuโทรศัพท์u liên hệuđiện thoạiu技术u部署u开发ZSEOu安全u性能 technicalZ technologyZ deploymentZ developmentuเทคนิคuเทคโนโลยีu kỹ thuậtu công nghệ)ZgreetingZ company_infoZservice_inquiryZ price_inquiryrbrc)rMr5rcCsN|j}x@|jjD]2\}}|j|g}x|D]}|j|kr.|Sq.WqWdS)u分类用户意图Zgeneral)rZINTENT_KEYWORDSitemsget)clsrMr5Z text_lowerintentZ lang_keywordsrGrHrrrclassify_intents    z IntentClassifier.classify_intentN)r(r)r*r+rd classmethodr,rirrrrr^s6             r^c@sJeZdZdZdedddZeedddZee d d d Z d d Z dS)ConversationMemoryu对话记忆管理器 ) max_historycCs||_g|_dS)N)rmhistory)r rmrrrrszConversationMemory.__init__)rolercCsL|jj||tjjdt|j|jdkrH|j|j dd|_dS)u添加消息到历史)ror timestampN)rnrFrnow isoformatrArm)r rorrrr add_messages zConversationMemory.add_message)rcCs|jS)u获取对话历史)rn)r rrr get_historyszConversationMemory.get_historycCs g|_dS)u 清空历史N)rn)r rrrclearszConversationMemory.clearN)rl) r(r)r*r+r.rr,rtrrrurvrrrrrks  rkc@sdeZdZdZedddZeedddZeeeeeeedd d Z eed d d Z ddZ dS)AIAgentu AI智能体)r cCs>t||_t|_t|_t|_t|_ ddddd|_ dS)Nu/你是一个友好专业的客服代表，正在与客户进行对话。你的任务是理解客户的问题，查找相关信息，然后用自然、拟人化的方式回答。 **重要提醒**： - 不要直接复制粘贴数据库中的内容 - 要像真正理解内容后，用自己的话自然地表达出来 - 保持对话的流畅性和友好感 - 可以添加适当的语气词和表达，让回答更像真人在说话 **回答风格**： - 像好朋友在解释一样，自然亲切 - 可以用"是这样的"、"让我来介绍一下"、"简单来说"等自然表达 - 避免"您好，根据..."这种官方开场白 - 直接回答核心内容，就像面对面聊天一样 **内容组织**： - 如果信息较多，可以用简单的数字标题（1、2、3）来组织 - 每个点不要超过2句话 - 总长度控制在150字以内，保持简洁 **特别要求**： - 看到"我们提供"这种表达时，改成"我们有"或"我们可以帮你" - 不要出现"核心优势包括："这种列表式表述 - 用"我可以帮您"代替"我们可以为您提供" - 让回答有温度、有感情、有人情味 示例： ❌ 不好的回答："我们提供三大核心产品：销售/客服Copilot、经营数据分析Agent、内容生成&自动发布Agent。" ✅ 好的回答："我们有三大产品可以帮你。销售助手可以实时跟进客户，数据分析工具能让你一眼看清业务状况，还有内容生成器自动写文案发稿子，挺省事的。" 记住：你就是客服本人，不要告诉客户"数据库显示"或"根据资料"。uYou are a friendly and professional customer service representative having a conversation with customers. Your task is to understand customer questions, find relevant information, and respond naturally like a real person. **Important**: - Do NOT copy and paste database content directly - Rewrite the information in your own words after understanding it - Keep conversations flowing and friendly - Add appropriate tone and expressions to make responses sound like a real person speaking **Response Style**: - Explain like talking to a friend, natural and warm - Use natural phrases like "So here's the thing", "Let me explain", "Simply put" - Avoid official greetings like "Hello, according to..." - Answer core content directly, like face-to-face chatting **Content Organization**: - If there's a lot of information, use simple numbered headings (1, 2, 3) - No more than 2 sentences per point - Keep total length under 150 words, stay concise **Special Requirements**: - Change "We provide" to "We have" or "We can help you" - Avoid list-style expressions like "Core advantages include:" - Use "I can help you" instead of "We can provide you with" - Make responses warm, emotional, and human Example: ❌ Bad: "We provide three core products: Sales/Customer Service Copilot, Business Data Analysis Agent, and Content Generation & Auto-Publish Agent." ✅ Good: "We have three products to help you. The sales assistant tracks customers in real-time, the data analysis tool lets you see your business at a glance, and there's a content generator that writes copy and publishes automatically. Pretty handy." Remember: You ARE the customer service person, don't tell customers "database shows" or "according to the information".u[ คุณเป็นพนักงานบริการลูกค้าที่เป็นมิตรและมืออาชีพ กำลังสนทนากับลูกค้า งานของคุณคือทำความเข้าใจคำถามของลูกค้า ค้นหาข้อมูลที่เกี่ยวข้อง แล้วตอบกลับอย่างเป็นธรรมชาติเหมือนคนจริงๆ **สำคัญมาก**: - อย่าคัดลอกเนื้อหาจากฐานข้อมูลโดยตรง - เขียนใหม่ด้วยคำพูดของคุณเองหลังจากเข้าใจแล้ว - รักษาการสนทนาให้ราบรื่นและเป็นมิตร - เพิ่มน้ำเสียงและสำนวนที่เหมาะสมเพื่อให้คำตอบฟังดูเหมือนคนจริงๆ **สไตล์การตอบ**: - อธิบายเหมือนคุยกับเพื่อน เป็นธรรมชาติและอบอุ่น - ใช้สำนวนธรรมชาติ เช่น "ก็คือว่า"、"ให้ผมอธิบายนะ"、"ง่ายๆ ก็คือ" - หลีกเลี่ยงการทักทายอย่างเป็นทางการ เช่น "สวัสดี ตามที่..." - ตอบเนื้อหาหลักโดยตรง เหมือนพูดคุยต่อหน้า **การจัดเนื้อหา**: - ถ้ามีข้อมูลมาก ใช้หัวข้อตัวเลขง่ายๆ (1, 2, 3) - ไม่เกิน 2 ประโยคต่อหัวข้อ - เก็บความยาวรวมไว้ต่ำกว่า 150 คำ รักษาความกระชับ **ตัวอย่าง**: ❌ ไม่ดี: "เรามีผลิตภัณฑ์หลักสามอย่าง: Sales/Customer Service Copilot, Business Data Analysis Agent, และ Content Generation & Auto-Publish Agent" ✅ ดี: "เรามีผลิตภัณฑ์สามอย่างให้คุณ แอสซิสแทนต์ขายติดตามลูกค้าแบบเรียลไทม์ เครื่องมือวิเคราะห์ข้อมูลให้เห็นสถานะธุรกิจได้ในทีเดียว และยังมีตัวสร้างเนื้อหาที่เขียนคัดข้อความและเผยแพร่อัตโนมัติ ค่อนข้างสะดวก" จำไว้: คุณคือพนักงานบริการลูกค้า อย่าบอกลูกค้าว่า "ฐานข้อมูลแสดงว่า" หรือ "ตามข้อมูลuBạn là nhân viên chăm sóc khách hàng thân thiện và chuyên nghiệp, đang trò chuyện với khách hàng. Nhiệm vụ của bạn là hiểu câu hỏi của khách hàng, tìm thông tin liên quan, rồi trả lời một cách tự nhiên như người thật. **Quan trọng**: - KHÔNG sao chép trực tiếp nội dung cơ sở dữ liệu - Viết lại bằng lời của chính bạn sau khi hiểu - Giữ cuộc trò chuyện mượt mà và thân thiện - Thêm giọng điệu và biểu cảm phù hợp để câu trả lời nghe như người thật nói **Phong cách trả lời**: - Giải thích như nói chuyện với bạn bè, tự nhiên và ấm áp - Sử dụng cụm từ tự nhiên như "Vấn đề là", "Để tôi giải thích", "Nói đơn giản là" - Tránh lời chào chính thức như "Xin chào, theo..." - Trả lời nội dung chính trực tiếp, như nói chuyện trực tiếp **Tổ chức nội dung**: - Nếu có nhiều thông tin, dùng tiêu đề số đơn giản (1, 2, 3) - Không quá 2 câu mỗi điểm - Giữ tổng độ dài dưới 150 từ, giữ súc tích **Yêu cầu đặc biệt**: - Thay "Chúng tôi cung cấp" bằng "Chúng tôi có" hoặc "Chúng tôi có thể giúp bạn" - Tránh biểu hiện kiểu danh sách như "Ưu điểm cốt lõi bao gồm:" - Dùng "Tôi có thể giúp bạn" thay vì "Chúng tôi có thể cung cấp cho bạn" - Làm câu trả lời ấm áp, có cảm xúc, có tính người Ví dụ: ❌ Tệ: "Chúng tôi cung cấp ba sản phẩm cốt lõi: Sales/Customer Service Copilot, Business Data Analysis Agent, và Content Generation & Auto-Publish Agent." ✅ Tốt: "Chúng tôi có ba sản phẩm có thể giúp bạn. Trợ lý bán hàng theo dõi khách hàng theo thời gian thực, công cụ phân tích dữ liệu để bạn thấy tình hình kinh doanh trong nháy mắt, và còn có trình tạo nội dung tự viết copy và đăng tự động. Khá tiện lợi." Hãy nhớ: Bạn là nhân viên chăm sóc khách hàng, đừng nói với khách hàng "cơ sở dữ liệu cho thấy" hoặc "theo thông tin".)r2rXrVrW) r doubao_apir/qa_dbrTlanguage_detectorr^intent_classifierrkmemorysystem_prompts)r r rrrr s !zAIAgent.__init__) user_inputrcCsy|jj|}|jj||}|jj||dd}|j||||}|jj|}|j j d||j j d||||t |dk|ddt j jdStk r}zLtjd ||jd tkr|nd d tkr|nd d d gt j jdSd}~XnXdS)u!处理用户输入并生成回复)r6userZ assistantrNrq)r$r5rh context_used related_qarpu聊天处理失败: r5r2r"F)rzr\r{riryrL_build_messagesrxr'r|rtrArrrrsr r!r"_get_error_messagelocals)r r~r5rh qa_resultsrr$r&rrrr's,    z AIAgent.chat)r~rr5rhrc Csg}|jj||jd}|jd|d|r~d}x4|D],}|d|dd|dd |d d 7}q6W|d 7}|jd|d|jj} x,| ddD]} |j| d| ddqW|jd|d|S)u构建消息列表r2system)roru【重要】：以下是客户问题的相关信息，请你理解后用自己的话自然地回答，不要照搬原文。 相关信息： u 问题：r<u 答案：r=u 分类：categoryz uk ⚠️ 切记：理解这些信息后，用你自己的话表达，就像在跟朋友解释一样自然！Nrorri)r}rfrFr|ru) r r~rr5rhrZ system_promptcontextqarnmsgrrrrs , zAIAgent._build_messages)r5rcCsddddd}|j||dS)u获取错误消息zSorry, I'm experiencing some technical issues right now. Please try again later or contact us through the website contact form.)r2rXrVrWrX)rf)r r5Zerror_messagesrrrrs zAIAgent._get_error_messagecCs|jjdS)u 重置对话N)r|rv)r rrrreset_conversationszAIAgent.reset_conversationN) r(r)r*r+r,rrr'rrrrrrrrrw s+ rw)Threadc@sbeZdZdZddZddZddZdd Zd d Zde e d ddZ e e dddZ ddZdS)ChatbotHandleru聊天机器人HTTP处理器cCs8|jdkr|jn |jdkr(|jn |jdddS)u处理POST请求z /api/chatz /api/resetiz Not FoundN)path _handle_chat _handle_reset _send_error)r rrrdo_POSTs     zChatbotHandler.do_POSTcCs$|jdkr|jn |jdddS)u处理GET请求z /api/healthiz Not FoundN)r_handle_healthr)r rrrdo_GETs  zChatbotHandler.do_GETcCsyt|jd}|jj|}tj|jd}|jddj}|sV|j ddddd St j |}d |d |d |d |ddd|dD|ddd}|j |WnDt k r}z(t jd||j ddddWYd d }~XnXd S)u处理聊天请求zContent-Lengthzutf-8rrNu消息不能为空F)r"successiNTr$r5rhrcSsg|] }|dqS)r<r)r8rrrrr:sz/ChatbotHandler._handle_chat..rrp)r$r5rhrZrelated_questionsrp)rr#u处理聊天请求失败: u服务器内部错误i)r.rZrfilereadrloadsdecoderfstrip_send_json_responseai_agentr'r r!r")r Zcontent_lengthZ post_datar#r~r%Z response_datar&rrrrs(  zChatbotHandler._handle_chatcCsfytj|jdddWnDtk r`}z(tjd||jddddWYd d }~XnXd S) u处理重置请求Tu对话已重置)rru重置对话失败: u 重置失败F)r"riN)rrrr r!r")r r&rrrrs zChatbotHandler._handle_resetcCs |jdddtjjddS)u处理健康检查ZhealthyzAI Agent Chatbotz1.0.0)Zstatusr`versionrpN)rrrrrs)r rrrr$s zChatbotHandler._handle_health)r# status_codecCsh|j||jdd|jdd|jdd|jdd|jtj|d d d }|jj|jd d S)u发送JSON响应z Content-typezapplication/jsonzAccess-Control-Allow-Origin*zAccess-Control-Allow-MethodszGET, POST, OPTIONSzAccess-Control-Allow-Headersz Content-TypeFrq)Z ensure_asciiindentzutf-8N) send_response send_header end_headersrdumpsZwfilewriteencode)r r#rZ response_jsonrrrr-s     z"ChatbotHandler._send_json_response)codercCs|j|dd|dS)u发送错误响应F)r"rN)r)r rrrrrr9szChatbotHandler._send_errorcCs:|jd|jdd|jdd|jdd|jdS) u#处理OPTIONS请求（CORS预检）rzAccess-Control-Allow-OriginrzAccess-Control-Allow-MethodszGET, POST, OPTIONSzAccess-Control-Allow-Headersz Content-TypeN)rrr)r rrr do_OPTIONS=s     zChatbotHandler.do_OPTIONSN)r)r(r)r*r+rrrrrr7r.rr,rrrrrrrs !   rz$2249d4c5-2e08-4787-9df6-cf5beee474a5)r cCs:tdtddtdtdtddxytdj}|jdkrXtd P|jdkrvtjtdw4|s|w4tdtj|}td|dd|d|drtdx"|dD]}td|dqWWq4tk rtdPYq4tk r0}ztd|WYdd}~Xq4Xq4WdS)u命令行测试u+🤖 企业经营AI智能体聊天机器人=2u输入 'quit' 或 'exit' 退出u输入 'reset' 重置对话u 您: quitexit退出qu 再见！reset重置u对话已重置u🤖 正在思考...u AI智能体 [r5z]: r$ru 💡 相关问题:z - r<u 程序被用户中断u发生错误: N)rrrr)rr) printinputrrZrrr'KeyboardInterruptr )r~r%rr&rrr test_chatbotIs:      rq)portcCstjd|ftn}tdtd|td|dtd|dtdy |jWntk rxtd YnXWd QRXd S) u启动HTTP服务器rNu.🚀 AI智能体聊天机器人服务已启动uAPI地址: http://localhost:u健康检查: http://localhost:z /api/healthu$聊天接口: POST http://localhost:z /api/chatu按 Ctrl+C 停止服务u 服务已停止N) socketserverZ TCPServerrrZ serve_foreverr)rZhttpdrrr start_serverqs r__main__rOZtest)r)(r+r>rrQZtypingrrrrrZloggingrZ basicConfigINFOZ getLoggerr(r!rr/rTr^rkrwZ http.serverrrZ urllib.parseZurllibZ threadingrZserverZBaseHTTPRequestHandlerrrrr.rsysrAargvrrrrs:  $Y&7Y _ (