Skip to content

Commit 451e751

Browse files
authored
Merge pull request #34 from xinhai-ai/main
修复0.5.9版本下的报错
2 parents 482ad8b + 30ccdd1 commit 451e751

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

resources/functions/openwebui_monitor.py

+11-2
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,15 @@ def _prepare_user_dict(self, __user__: dict) -> dict:
7676
user_dict["valves"] = user_dict["valves"].model_dump()
7777

7878
return user_dict
79+
def _prepare_body_dict(self, body: dict) -> dict:
80+
"""将 body 对象转换为可序列化的字典"""
81+
body_dict = dict(body)
82+
if "model" in body_dict["metadata"] and hasattr(
83+
body_dict["metadata"]["model"], "model_dump"
84+
):
85+
body_dict["metadata"]["model"] = body_dict["metadata"]["model"].model_dump()
86+
87+
return body_dict
7988

8089
def inlet(
8190
self, body: dict, user: Optional[dict] = None, __user__: dict = {}
@@ -87,9 +96,9 @@ def inlet(
8796
headers = {"Authorization": f"Bearer {self.valves.API_KEY}"}
8897

8998
user_dict = self._prepare_user_dict(__user__)
90-
99+
body_dict = self._prepare_body_dict(body)
91100
response = requests.post(
92-
post_url, headers=headers, json={"user": user_dict, "body": body}
101+
post_url, headers=headers, json={"user": user_dict, "body": body_dict}
93102
)
94103

95104
if response.status_code == 401:

0 commit comments

Comments
 (0)