api

type ChatPromptBuilder interface {
	/*
		CreatePrompt build a complete prompt using the existing session exchanges, and the new
		request from the user.

			@param ctxt context.Context - query context
			@param session persistence.ChatSession - current chat session
			@param newRequest string - new user request
			@return complete prompt for the text completion model
	*/
	CreatePrompt(
		ctxt context.Context, session persistence.ChatSession, newRequest string,
	) (string, error)
}

type ChatSessionHandler interface {
	/*
		SendRequest send a new request within the session

			@param ctxt context.Context - query context
			@param prompt string - the prompt to send
			@param resp chan string - channel for sending out the responses from the model
	*/
	SendRequest(ctxt context.Context, prompt string, resp chan string) error

	/*
		Close close this session

			@param ctxt context.Context - query context
	*/
	Close(ctxt context.Context) error
}

type Client interface {
	/*
		MakeCompletionRequest make a completion request to the model

			@param ctxt context.Context - query context
			@param session persistence.ChatSession - chat session parameters
			@param prompt string - the prompt to send
			@param resp chan string - channel for sending out the responses from the model
	*/
	MakeCompletionRequest(
		ctxt context.Context, session persistence.ChatSession, prompt string, resp chan string,
	) error
}