Put logging of conversations behind ENABLE_CONVERSATIONS_LOGGING

This commit is contained in:
Ettore Di Giacinto
2025-03-04 22:23:58 +01:00
parent d288755444
commit 2942668d89
2 changed files with 10 additions and 2 deletions

View File

@@ -72,12 +72,18 @@ func NewAgentPool(
connectors func(*AgentConfig) []Connector,
promptBlocks func(*AgentConfig) []PromptBlock,
timeout string,
withLogs bool,
) (*AgentPool, error) {
// if file exists, try to load an existing pool.
// if file does not exist, create a new pool.
poolfile := filepath.Join(directory, "pool.json")
conversationPath := ""
if withLogs {
conversationPath = filepath.Join(directory, "conversations")
}
if _, err := os.Stat(poolfile); err != nil {
// file does not exist, create a new pool
return &AgentPool{
@@ -95,7 +101,7 @@ func NewAgentPool(
availableActions: availableActions,
promptBlocks: promptBlocks,
timeout: timeout,
conversationLogs: filepath.Join(directory, "conversations"),
conversationLogs: conversationPath,
}, nil
}
@@ -118,7 +124,7 @@ func NewAgentPool(
promptBlocks: promptBlocks,
availableActions: availableActions,
timeout: timeout,
conversationLogs: filepath.Join(directory, "conversations"),
conversationLogs: conversationPath,
}, nil
}

View File

@@ -16,6 +16,7 @@ var apiKey = os.Getenv("API_KEY")
var timeout = os.Getenv("TIMEOUT")
var stateDir = os.Getenv("STATE_DIR")
var localRAG = os.Getenv("LOCAL_RAG")
var withLogs = os.Getenv("ENABLE_CONVERSATIONS_LOGGING") == "true"
func init() {
if testModel == "" {
@@ -52,6 +53,7 @@ func main() {
services.Connectors,
services.PromptBlocks,
timeout,
withLogs,
)
if err != nil {
panic(err)