接入模型通用工具调用与客服语义理解

2026-11-06 16:35:52 +08:00
parent afe2bd1207
commit bca0a6914a
14 changed files with 752 additions and 14 deletions
@@ -0,0 +1,97 @@
+"""Execute whitelisted model requests under the current server identity."""
+
+import json
+import re
+
+from .customer_service import order_login_response
+from .visitor_service import customer_tenant_ids
+
+
+def validate_calls(message):
+    from .model_prompts import TOOLS, PARAMETERS
+
+    calls = message.get('tool_calls')
+    if not isinstance(calls, list) or not 1 <= len(calls) <= 4:
+        raise ValueError('Invalid tool count')
+    names = {tool['function']['name'] for tool in TOOLS}
+    requests = []
+    ids = set()
+    for call in calls:
+        function = call.get('function', {})
+        call_id = call.get('id')
+        if call.get('type') != 'function' or function.get('name') not in names:
+            raise ValueError('Unknown tool')
+        if not isinstance(call_id, str) or not 1 <= len(call_id) <= 200 or call_id in ids:
+            raise ValueError('Invalid call ID')
+        arguments = json.loads(function.get('arguments', ''))
+        name = function['name']
+        if not isinstance(arguments, dict) or set(arguments) != {'question', *PARAMETERS[name]}:
+            raise ValueError('Invalid tool arguments')
+        question = arguments['question']
+        if not isinstance(question, str) or not 0 < len(question.strip()) <= 2000:
+            raise ValueError('Invalid tool question')
+        for key, schema in PARAMETERS[name].items():
+            value = arguments[key]
+            if 'enum' in schema and value not in schema['enum']:
+                raise ValueError('Invalid tool choice')
+            if key == 'services' and (not isinstance(value, list) or len(value) > 12 or
+                    any(not isinstance(service, str) or not 0 < len(service) <= 100 for service in value)):
+                raise ValueError('Invalid channels')
+            if key == 'reference' and (not isinstance(value, str) or len(value) > 64 or
+                    value and not re.fullmatch(r'[A-Za-z0-9_-]+', value)):
+                raise ValueError('Invalid order reference')
+        if name == 'query_orders' and ((arguments['condition'] == 'stage' and not arguments['stage']) or
+                (arguments['condition'] == 'status' and not arguments['status'])):
+            raise ValueError('Missing order condition')
+        ids.add(call_id)
+        requests.append((call_id, name, {**arguments, 'question': question.strip()}))
+    return requests
+
+
+def execute_tool(name, arguments, conv_id):
+    question = arguments['question']
+    if name == 'query_orders':
+        if not customer_tenant_ids():
+            return order_login_response()
+        from .order_data import answer_query
+        if arguments['action'] == 'lookup':
+            from .order_data import IDENTIFIER
+            if not arguments['reference']:
+                return {'t': '请提供单个入仓码查询实际物流轨迹。'}
+            question = arguments['reference'] + ' ' + IDENTIFIER.sub('', question)
+        if arguments['action'] in ('count', 'list'):
+            from .order_data import STATUS_LABELS
+            from .order_query import STAGE_LABELS
+            condition = arguments['condition']
+            if condition == 'unsupported':
+                return {'t': '当前尚不支持该筛选条件，请提供入仓码逐票查看实际轨迹。'}
+            if arguments['action'] == 'count':
+                from .order_listing import answer_summary
+                query = {'status': arguments['status'] or None, 'stage': arguments['stage'] or None,
+                         'exception': condition == 'exception', 'grouped': condition == 'grouped', 'unsupported': False}
+                return {**answer_summary(query, customer_tenant_ids(), conv_id), '_customer_data': True}
+            label = ('异常' if condition == 'exception' else '')
+            label += STAGE_LABELS.get(arguments['stage'], '') + STATUS_LABELS.get(arguments['status'], '')
+            question = '列出我的' + label + '订单'
+        answer = answer_query(question, conv_id)
+        if answer is None:
+            answer = {'t': '请提供入仓码，或说明要查询的订单统计条件。'}
+        return {**answer, '_customer_data': True}
+    if name == 'quote_shipping':
+        from .quote_service import answer_quote
+        from .quote_scope import answer_us_fallback
+        request = {key: arguments[key] for key in ('request', 'selection', 'services', 'policy_topic')}
+        return (answer_quote(question, conv_id, request=request) or answer_us_fallback(question, conv_id)
+                or {'t': '请提供目的国家、货物重量及目的仓码或邮编，继续核对当前资料。'})
+    if name == 'read_service_info':
+        from .customer_faq import answer_capability
+        from .knowledge import search_knowledge
+        answer = answer_capability(question)
+        if answer is not None:
+            return answer
+        # Only service/contact information is universal across the old regional knowledge base.
+        if any(word in question for word in ('人工', '客服', '工作时间', '营业时间', '联系方式')):
+            return {'t': search_knowledge('人工客服工作时间')}
+        return {'t': '可以咨询物流发货、报价、订单和轨迹。具体订单需登录查询，具体运费需按货物条件查询报价工具。'
+                     '本次没有查得其他业务事实。可以回应问候、询问客户想办什么，或针对原问题澄清缺失条件。'}
+    raise ValueError('Unknown tool')