OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-04 03:37:35 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-04-10 21:22:14 -07:00
parent b97ea72616
commit 6d63337ee9
2 changed files with 31 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
backend/open_webui/utils/tools.py
View File

@ -6,12 +6,28 @@ import aiohttp
import asyncio import asyncio
import yaml import yaml
from typing import Any, Awaitable, Callable, get_type_hints, Dict, List, Union, Optional from pydantic import BaseModel
from pydantic.fields import FieldInfo
from typing import (
Any,
Awaitable,
Callable,
get_type_hints,
get_args,
get_origin,
Dict,
List,
Tuple,
Union,
Optional,
Type,
)
from functools import update_wrapper, partial from functools import update_wrapper, partial
from fastapi import Request from fastapi import Request
from pydantic import BaseModel, Field, create_model from pydantic import BaseModel, Field, create_model
from langchain_core.utils.function_calling import ( from langchain_core.utils.function_calling import (
convert_to_openai_function as convert_pydantic_model_to_openai_function_spec, convert_to_openai_function as convert_pydantic_model_to_openai_function_spec,
) )
@ -259,22 +275,25 @@ def convert_function_to_pydantic_model(func: Callable) -> type[BaseModel]:
parameters = signature.parameters parameters = signature.parameters
docstring = func.__doc__ docstring = func.__doc__
descriptions = parse_docstring(docstring)
tool_description = parse_description(docstring) description = parse_description(docstring)
function_descriptions = parse_docstring(docstring)
field_defs = {} field_defs = {}
for name, param in parameters.items(): for name, param in parameters.items():
type_hint = type_hints.get(name, Any) type_hint = type_hints.get(name, Any)
default_value = param.default if param.default is not param.empty else ... default_value = param.default if param.default is not param.empty else ...
description = descriptions.get(name, None)
if not description: description = function_descriptions.get(name, None)
if description:
field_defs[name] = type_hint, Field(default_value, description=description)
else:
field_defs[name] = type_hint, default_value field_defs[name] = type_hint, default_value
continue
field_defs[name] = type_hint, Field(default_value, description=description)
model = create_model(func.__name__, **field_defs) model = create_model(func.__name__, **field_defs)
model.__doc__ = tool_description model.__doc__ = description
return model return model
@ -300,11 +319,13 @@ def get_tool_specs(tool_module: object) -> list[dict]:
convert_function_to_pydantic_model, get_functions_from_tool(tool_module) convert_function_to_pydantic_model, get_functions_from_tool(tool_module)
) )
return [ specs = [
convert_pydantic_model_to_openai_function_spec(function_model) convert_pydantic_model_to_openai_function_spec(function_model)
for function_model in function_models for function_model in function_models
] ]
return specs
def resolve_schema(schema, components): def resolve_schema(schema, components):
""" """

2
src/lib/components/chat/Messages/UserMessage.svelte
View File

@ -192,7 +192,7 @@
<div> <div>
<button <button
id="save-edit-message-button" id="save-edit-message-button"
class=" px-4 py-2 bg-gray-50 hover:bg-gray-100 dark:bg-gray-800 dark:hover:bg-gray-700 border dark:border-gray-700 text-gray-700 dark:text-gray-200 transition rounded-3xl" class=" px-4 py-2 bg-gray-50 hover:bg-gray-100 dark:bg-gray-800 dark:hover:bg-gray-700 border border-gray-100 dark:border-gray-700 text-gray-700 dark:text-gray-200 transition rounded-3xl"
on:click={() => { on:click={() => {
editMessageConfirmHandler(false); editMessageConfirmHandler(false);
}} }}