huipeng6
commited on
Commit
•
14c9582
1
Parent(s):
49ab6fa
feat: 适配function-calling
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from huggingface_hub import InferenceClient
|
|
4 |
"""
|
5 |
For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
|
6 |
"""
|
7 |
-
client = InferenceClient("
|
8 |
|
9 |
|
10 |
def respond(
|
@@ -21,7 +21,15 @@ def respond(
|
|
21 |
if val[0]:
|
22 |
messages.append({"role": "user", "content": val[0]})
|
23 |
if val[1]:
|
24 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
|
26 |
messages.append({"role": "user", "content": message})
|
27 |
|
|
|
4 |
"""
|
5 |
For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
|
6 |
"""
|
7 |
+
client = InferenceClient("xalss/Qwen2-7B-Instruct-glaive-function-calling")
|
8 |
|
9 |
|
10 |
def respond(
|
|
|
21 |
if val[0]:
|
22 |
messages.append({"role": "user", "content": val[0]})
|
23 |
if val[1]:
|
24 |
+
assistant = val[1]
|
25 |
+
messages.append({
|
26 |
+
"role": (
|
27 |
+
"function"
|
28 |
+
if assistant.startswith("<FUNCTION RESPONSE>")
|
29 |
+
else "assistant"
|
30 |
+
),
|
31 |
+
"content": assistant.lstrip("<FUNCTION RESPONSE>"),
|
32 |
+
})
|
33 |
|
34 |
messages.append({"role": "user", "content": message})
|
35 |
|