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 GetTheradAndRunResponse struct {
	ID             string        `json:"id"`
	Object         string        `json:"object"`
	CreatedAt      int           `json:"created_at"`
	AssistantId    string        `json:"assistant_id"`
	ThreadId       string        `json:"thread_id"`
	Status         string        `json:"status"`
	StartedAt      interface{}   `json:"started_at"`
	ExpiresAt      int           `json:"expires_at"`
	CancelledAt    interface{}   `json:"cancelled_at"`
	FailedAt       interface{}   `json:"failed_at"`
	CompletedAt    interface{}   `json:"completed_at"`
	RequiredAction interface{}   `json:"required_action"`
	LastError      interface{}   `json:"last_error"`
	Model          string        `json:"model"`
	Instructions   string        `json:"instructions"`
	Tools          []interface{} `json:"tools"`
	ToolResources  struct {
	} `json:"tool_resources"`
	Metadata struct {
	} `json:"metadata"`
	Temperature         float64     `json:"temperature"`
	TopP                float64     `json:"top_p"`
	MaxCompletionTokens interface{} `json:"max_completion_tokens"`
	MaxPromptTokens     interface{} `json:"max_prompt_tokens"`
	TruncationStrategy  struct {
		Type         string      `json:"type"`
		LastMessages interface{} `json:"last_messages"`
	} `json:"truncation_strategy"`
	IncompleteDetails interface{} `json:"incomplete_details"`
	Usage             interface{} `json:"usage"`
	ResponseFormat    struct {
		Type string `json:"type"`
	} `json:"response_format"`
	ToolChoice        string `json:"tool_choice"`
	ParallelToolCalls bool   `json:"parallel_tool_calls"`
}

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"`
}