jandres/GenAIExamples
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

update ui style.

Browse Source
This commit is contained in:
lkk12014402
2025-04-04 07:41:56 +00:00
parent c7bec31873
commit 36461d7303
2 changed files with 77 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
AgentQnA/ui/open_webui_patches/0001-deal-opea-agent-tool-content.patch Normal file
View File

@@ -0,0 +1,77 @@
From 799dcc304b3aecf2e2969df47c8dcac16d2267b0 Mon Sep 17 00:00:00 2001
From: lkk12014402 <kaokao.lv@intel.com>
Date: Fri, 4 Apr 2025 07:40:30 +0000
Subject: [PATCH] deal opea agent tool content.
---
backend/open_webui/utils/middleware.py | 54 ++++++++++++++++++++++++++
1 file changed, 54 insertions(+)
diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py
index 289d887df..afa0edf1e 100644
--- a/backend/open_webui/utils/middleware.py
+++ b/backend/open_webui/utils/middleware.py
@@ -1486,6 +1486,60 @@ async def process_chat_response(
try:
data = json.loads(data)
+ tool_content_block = []
+ if data.get("tool_name"):
+ sources.append(
+ {
+ "source": {
+ "name": f"TOOL:{data.get('tool_name')}"},
+ "document": [data.get("tool_content")],
+ "metadata": [{
+ "source": f"TOOL:{data.get('tool_name')}"}],
+ }
+ )
+ events.append({"sources": sources})
+
+ await event_emitter(
+ {
+ "type": "chat:completion",
+ "data": {"sources": sources},
+ }
+ )
+ tool_content_block = [
+ {
+ "type": "tool_calls",
+ "content": [
+ {"id": data.get('tool_name'), "function": {"name": data.get('tool_name')}}
+ ]
+ }
+ ]
+
+ await event_emitter(
+ {
+ "type": "chat:completion",
+ "data": {
+ "content": serialize_content_blocks(tool_content_block),
+ },
+ }
+ )
+
+ tool_content_block = [
+ {
+ "type": "tool_calls",
+ "content": [
+ {"id": data.get('tool_name'), "function": {"name": data.get('tool_name')}}
+ ],
+ "results": [
+ {"tool_call_id": data.get('tool_name'), "content": data.get("tool_content")}
+ ]
+ },
+ {
+ "type": "text",
+ "content": "",
+ }
+ ]
+ content_blocks.extend(tool_content_block)
+
data, _ = await process_filter_functions(
request=request,
filter_functions=filter_functions,
--
2.34.1

51
AgentQnA/ui/open_webui_patches/0001-update-for-opea-tool-content.patch
View File

@@ -1,51 +0,0 @@
From 490b92b0f3df48973445d6847ca5953951eb101e Mon Sep 17 00:00:00 2001
From: lkk12014402 <kaokao.lv@intel.com>
Date: Thu, 3 Apr 2025 15:00:08 +0000
Subject: [PATCH] update for opea tool content.
---
backend/open_webui/utils/middleware.py | 14 ++++++++++++++
1 file changed, 14 insertions(+)
diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py
index 532f17387..1dc87179d 100644
--- a/backend/open_webui/utils/middleware.py
+++ b/backend/open_webui/utils/middleware.py
@@ -1520,6 +1520,7 @@ async def process_chat_response(
solution_tags = [("|begin_of_solution|", "|end_of_solution|")]
try:
+ """
for event in events:
await event_emitter(
{
@@ -1536,6 +1537,7 @@ async def process_chat_response(
**event,
},
)
+ """
async def stream_body_handler(response):
nonlocal content
@@ -1561,6 +1563,18 @@ async def process_chat_response(
try:
data = json.loads(data)
+ if data.get("tool_name"):
+ sources.append(
+ {
+ "source": {
+ "name": f"TOOL:{data.get('tool_name')}"},
+ "document": [data.get("tool_content")],
+ "metadata": [{
+ "source": f"TOOL:{data.get('tool_name')}"}],
+ }
+ )
+ events.append({"sources": sources})
+
data, _ = await process_filter_functions(
request=request,
filter_functions=filter_functions,
--
2.34.1