openAIAssistantRunnable

type AddMessageResponse struct {
	ID string `json:"id"`
}

type AgentExecutor struct {
	Agent *Assistant
	Tools []tools.Tool
	// contains filtered or unexported fields
}

type Assistant struct {
	ID     string
	Name   string
	APIKey string
	Model  string
	Tools  []tools.Tool
}

type CreateAssistantRequest struct {
	Instructions string      `json:"instructions"`
	Name         string      `json:"name"`
	Tools        []llms.Tool `json:"tools"`
	Model        string      `json:"model"`
}

type CreateAssistantResponse struct {
	ID           string      `json:"id"`
	Name         string      `json:"name"`
	Model        string      `json:"model"`
	Instructions string      `json:"instructions"`
	Tools        []llms.Tool `json:"tools"`
}

type CreateRunRequest struct {
	AssistantID  string `json:"assistant_id"`
	Instructions string `json:"instructions"`
}

type CreateRunResponse struct {
	ID             string `json:"id"`
	Status         string `json:"status"`
	RequiredAction struct {
		SubmitToolOutputs struct {
			ToolCalls []ToolCall `json:"tool_calls"`
		} `json:"submit_tool_outputs"`
	} `json:"required_action"`
}

type CreateThreadAndRunRequest struct {
	AssistantID string `json:"assistant_id"`
	Thread      Thread `json:"thread"`
}

type CreateThreadResponse struct {
	ID string `json:"id"`
}

type EchoTool struct{}

type GetThreadMessagesResponse struct {
	Object string `json:"object"`
	Data   []struct {
		Id          string  `json:"id"`
		Object      string  `json:"object"`
		CreatedAt   int     `json:"created_at"`
		AssistantId *string `json:"assistant_id"`
		ThreadId    string  `json:"thread_id"`
		RunId       *string `json:"run_id"`
		Role        string  `json:"role"`
		Content     []struct {
			Type string `json:"type"`
			Text struct {
				Value       string        `json:"value"`
				Annotations []interface{} `json:"annotations"`
			} `json:"text"`
		} `json:"content"`
		Attachments []interface{} `json:"attachments"`
		Metadata    struct {
		} `json:"metadata"`
	} `json:"data"`
	FirstId string `json:"first_id"`
	LastId  string `json:"last_id"`
	HasMore bool   `json:"has_more"`
}

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type Option func(*Assistant)

type Thread struct {
	Messages []Message `json:"messages"`
}

type ToolCall struct {
	ID       string `json:"id"`
	Type     string `json:"type"`
	Function struct {
		Name      string `json:"name"`
		Arguments string `json:"arguments"`
	} `json:"function"`
}

type ToolConfig struct {
	Type     string                   `json:"type"`
	Function *llms.FunctionDefinition `json:"function,omitempty"`
}