chatmodels

type ChatModel string

type Client struct {
	*Resources
}

type CloudFlareAiWorkerAPI interface {
	GenerateText(context.Context, string, string) (string, error)
	GenerateImage(ctx context.Context, prompt string, model string) ([]byte, error)
}

type CloudflareApiClient struct {
	AccountID string
	APIKey    string
}

type GeminiAPI interface {
	GeminiChat(context.Context, string) (*genai.GenerateContentResponse, error)
}

type GeminiApiClient struct {
	// contains filtered or unexported fields
}

type GptAPI interface {
	AutoComplete(context.Context, string) (openai.ChatCompletionResponse, error)
}

type LastResponse struct {
	Prompt         string    `json:"prompt"`
	Response       string    `json:"response"`
	TimeDiff       string    `json:"time_diff"`
	Model          ChatModel `json:"model"`
	ImagesResponse []string  `json:"images_responses"`
	Error          string    `json:"error_message"`
}

type MockClient struct {
	Service
	mock.Mock
}

type OpenAIApiClient struct {
	Token        string
	OpenAIClient *openai.Client
}

type Request struct {
	Prompt string    `json:"prompt"`
	Model  ChatModel `json:"model"`
}

type Resources struct {
	GPTApi              GptAPI
	GeminiAPI           GeminiAPI
	CloudflareApiClient CloudFlareAiWorkerAPI
}

type Response struct {
	Result struct {
		Response string `json:"response"`
	} `json:"result,omitempty"`
	Errors []struct {
		Code    int    `json:"code"`
		Message string `json:"message"`
	} `json:"errors,omitempty"`
	Messages []string `json:"messages,omitempty"`
	Success  bool     `json:"success"`
}

type Service interface {
	AutoComplete(context.Context, string, ChatModel) (string, error)
	GenerateImage(context.Context, string, ChatModel) ([]byte, error)
}