Initiate agents from pool

agent/agent.go

@@ -39,10 +39,7 @@ type Agent struct {
 func New(opts ...Option) (*Agent, error) {
 	options, err := newOptions(opts...)
 	if err != nil {
-		if err != nil {
-			err = fmt.Errorf("failed to set options: %v", err)
-		}
-		return nil, err
+		return nil, fmt.Errorf("failed to set options: %v", err)
 	}
 
 	client := llm.NewClient(options.LLMAPI.APIKey, options.LLMAPI.APIURL)
@@ -126,6 +123,12 @@ func (a *Agent) CurrentConversation() []openai.ChatCompletionMessage {
 	return a.currentConversation
 }
 
+func (a *Agent) SetConversation(conv []openai.ChatCompletionMessage) {
+	a.Lock()
+	defer a.Unlock()
+	a.currentConversation = conv
+}
+
 func (a *Agent) ResetConversation() {
 	a.Lock()
 	defer a.Unlock()

agent/options.go

@@ -94,6 +94,20 @@ func WithLLMAPIURL(url string) Option {
 	}
 }
 
+func WithStateFile(path string) Option {
+	return func(o *options) error {
+		o.statefile = path
+		return nil
+	}
+}
+
+func WithCharacterFile(path string) Option {
+	return func(o *options) error {
+		o.characterfile = path
+		return nil
+	}
+}
+
 func WithLLMAPIKey(key string) Option {
 	return func(o *options) error {
 		o.LLMAPI.APIKey = key