rework llm

1 files changed, 49 insertions, 0 deletions

diff --git a/internal/llm/tools/tools.go b/internal/llm/tools/tools.go
new file mode 100644
index 000000000..e15c1c31f
--- /dev/null
+++ b/internal/llm/tools/tools.go
@@ -0,0 +1,49 @@
+package tools
+
+import "context"
+
+type ToolInfo struct {
+	Name        string
+	Description string
+	Parameters  map[string]any
+	Required    []string
+}
+
+type toolResponseType string
+
+const (
+	ToolResponseTypeText  toolResponseType = "text"
+	ToolResponseTypeImage toolResponseType = "image"
+)
+
+type ToolResponse struct {
+	Type    toolResponseType `json:"type"`
+	Content string           `json:"content"`
+	IsError bool             `json:"is_error"`
+}
+
+func NewTextResponse(content string) ToolResponse {
+	return ToolResponse{
+		Type:    ToolResponseTypeText,
+		Content: content,
+	}
+}
+
+func NewTextErrorResponse(content string) ToolResponse {
+	return ToolResponse{
+		Type:    ToolResponseTypeText,
+		Content: content,
+		IsError: true,
+	}
+}
+
+type ToolCall struct {
+	ID    string `json:"id"`
+	Name  string `json:"name"`
+	Input string `json:"input"`
+}
+
+type BaseTool interface {
+	Info() ToolInfo
+	Run(ctx context.Context, params ToolCall) (ToolResponse, error)
+}