Skip to content

Commit

Permalink
Don't set default prompt param anymore
Browse files Browse the repository at this point in the history
  • Loading branch information
ibuildthecloud committed Feb 8, 2024
1 parent 853c4c7 commit 912d1b4
Show file tree
Hide file tree
Showing 5 changed files with 47 additions and 44 deletions.
8 changes: 4 additions & 4 deletions pkg/builtin/builtin.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ import (
"github.com/jaytaylor/html2text"
)

var Tools = map[string]types.Tool{
var tools = map[string]types.Tool{
"sys.read": {
Description: "Reads the contents of a file",
Arguments: types.ObjectSchema(
Expand Down Expand Up @@ -94,7 +94,7 @@ func SysProgram() *types.Program {

func ListTools() (result []types.Tool) {
var keys []string
for k := range Tools {
for k := range tools {
keys = append(keys, k)
}

Expand All @@ -109,7 +109,7 @@ func ListTools() (result []types.Tool) {

func Builtin(name string) (types.Tool, bool) {
name, dontFail := strings.CutSuffix(name, "?")
t, ok := Tools[name]
t, ok := tools[name]
t.Name = name
t.ID = name
t.Instructions = "#!" + name
Expand All @@ -123,7 +123,7 @@ func Builtin(name string) (types.Tool, bool) {
return s, err
}
}
return t, ok
return SetDefaults(t), ok
}

func SysFind(ctx context.Context, env []string, input string) (string, error) {
Expand Down
22 changes: 22 additions & 0 deletions pkg/builtin/defaults.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
package builtin

import (
"github.com/gptscript-ai/gptscript/pkg/openai"
"github.com/gptscript-ai/gptscript/pkg/types"
)

var (
DefaultModel = openai.DefaultModel
DefaultVisionModel = openai.DefaultVisionModel
)

func SetDefaults(tool types.Tool) types.Tool {
if tool.ModelName == "" {
if tool.Vision {
tool.ModelName = DefaultVisionModel
} else {
tool.ModelName = DefaultModel
}
}
return tool
}
19 changes: 18 additions & 1 deletion pkg/engine/engine.go
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,19 @@ You do not need to explain the steps taken, only provide the result to the given
You are referred to as a tool.
`

var DefaultToolSchema = types.JSONSchema{
Property: types.Property{
Type: "object",
},
Properties: map[string]types.Property{
openai.DefaultPromptParameter: {
Description: "Prompt to send to the tool or assistant. This may be instructions or question.",
Type: "string",
},
},
Required: []string{openai.DefaultPromptParameter},
}

var completionID int64

func init() {
Expand Down Expand Up @@ -185,12 +198,16 @@ func (e *Engine) Start(ctx Context, input string) (*Return, error) {
if err != nil {
return nil, err
}
args := subTool.Arguments
if args == nil && !subTool.IsCommand() {
args = &DefaultToolSchema
}
completion.Tools = append(completion.Tools, types.CompletionTool{
Type: types.CompletionToolTypeFunction,
Function: types.CompletionFunctionDefinition{
Name: subToolName,
Description: subTool.Description,
Parameters: subTool.Arguments,
Parameters: args,
},
})
}
Expand Down
37 changes: 0 additions & 37 deletions pkg/loader/defaults.go

This file was deleted.

5 changes: 3 additions & 2 deletions pkg/loader/loader.go
Original file line number Diff line number Diff line change
Expand Up @@ -154,7 +154,8 @@ func loadProgram(data []byte, into *types.Program) (types.Tool, error) {
for tk, tv := range v.ToolMapping {
v.ToolMapping[tk] = tv + id
}
into.ToolSet[k+id] = v
v.ID = k + id
into.ToolSet[v.ID] = v
}

return into.ToolSet[ext.EntryToolID+id], nil
Expand Down Expand Up @@ -314,7 +315,7 @@ func link(ctx context.Context, prg *types.Program, base *Source, tool types.Tool
tool.Tools[i] = newToolName
}

tool = SetDefaults(tool)
tool = builtin.SetDefaults(tool)
prg.ToolSet[tool.ID] = tool

return tool, nil
Expand Down

0 comments on commit 912d1b4

Please sign in to comment.