alex/LocalAGI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

more logging

Browse Source
This commit is contained in:
Ettore Di Giacinto
2024-04-19 00:08:18 +02:00
parent 08563c3286
commit f6dcc09562
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
agent/agent.go
View File

@@ -96,6 +96,7 @@ func (a *Agent) StopAction() {
a.Lock() a.Lock()
defer a.Unlock() defer a.Unlock()
if a.actionContext != nil { if a.actionContext != nil {
xlog.Debug("Stopping current action", "agent", a.Character.Name)
a.actionContext.Cancel() a.actionContext.Cancel()
} }
} }
@@ -115,6 +116,11 @@ func (a *Agent) ConversationChannel() chan openai.ChatCompletionMessage {
// Ask is a pre-emptive, blocking call that returns the response as soon as it's ready. // Ask is a pre-emptive, blocking call that returns the response as soon as it's ready.
// It discards any other computation. // It discards any other computation.
func (a *Agent) Ask(opts ...JobOption) *JobResult { func (a *Agent) Ask(opts ...JobOption) *JobResult {
xlog.Debug("Agent is being asked", "agent", a.Character.Name)
defer func() {
xlog.Debug("Agent has finished being asked", "agent", a.Character.Name)
}()
a.StopAction() a.StopAction()
j := NewJob( j := NewJob(
append( append(
@@ -123,8 +129,9 @@ func (a *Agent) Ask(opts ...JobOption) *JobResult {
WithResultCallback(a.options.resultCallback), WithResultCallback(a.options.resultCallback),
)..., )...,
) )
// xlog.Info("Job created", text) xlog.Debug("Job created", "agent", a.Character.Name, "job", j)
a.jobQueue <- j a.jobQueue <- j
xlog.Debug("Waiting result", "agent", a.Character.Name, "job", j)
return j.Result.WaitResult() return j.Result.WaitResult()
} }