mirrors/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-05-10 18:06:33 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

wip?

Browse source
This commit is contained in:
ParthSareen 2025-05-07 19:00:44 -07:00
parent 611d3a17ed
commit 23e8ac9428
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
server/python_tools.go
View file

@ -155,9 +155,9 @@ func parsePythonValue(value string) (any, error) {
return nil, fmt.Errorf("invalid Python value: %s", value)
}
// parsePythonFunctionCall parses Python function calls from a string
// parsePythonToolCall parses Python function calls from a string
// it supports keyword arguments, as well as multiple functions in a single string
func parsePythonFunctionCall(s string) ([]api.ToolCall, error) {
func parsePythonToolCall(s string) ([]api.ToolCall, error) {
matches := pythonFuncRegex.FindAllStringSubmatchIndex(s, -1)
if len(matches) == 0 {
return nil, fmt.Errorf("no Python function calls found")
@ -167,7 +167,7 @@ func parsePythonFunctionCall(s string) ([]api.ToolCall, error) {
for _, match := range matches {
name := s[match[2]:match[3]]
args := s[match[4]:match[5]]
arguments := make(api.ToolCallFunctionArguments)
var arguments api.ToolCallFunctionArguments
if len(args) == 0 {
toolCalls = append(toolCalls, api.ToolCall{
Function: api.ToolCallFunction{

2
server/python_tools_test.go
View file

@ -143,7 +143,7 @@ func TestParsePythonFunctionCall(t *testing.T) {
for _, tt := range cases {
t.Run(tt.name, func(t *testing.T) {
got, err := parsePythonFunctionCall(tt.input)
got, err := parsePythonToolCall(tt.input)
if (err != nil) != tt.err {
t.Fatalf("expected error: %v, got error: %v", tt.err, err)
}