opencode/yao
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #821 from trheyi/main

Browse source 
Enhance Neo API assistant with improved error handling and message st…
This commit is contained in:
Max 2025-01-18 15:02:30 +08:00 • committed by GitHub
commit 883dd07fb0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
neo/assistant/hooks.go
View file

@ -221,7 +221,9 @@ func (ast *Assistant) call(ctx context.Context, method string, context chatctx.C
// Wait for either context cancellation or method completion // Wait for either context cancellation or method completion
select { select {
case <-ctx.Done(): case <-ctx.Done():
scriptCtx.Close() // Force close the script context if scriptCtx != nil {
scriptCtx.Close() // Force close the script context
}
return nil, ctx.Err() return nil, ctx.Err()
case <-done: case <-done:
return result, callErr return result, callErr

6
neo/message/content.go
View file

@ -52,6 +52,12 @@ func (c *Content) String() string {
data := map[string]interface{}{ data := map[string]interface{}{
"id": c.ID, "id": c.ID,
"type": "function", "type": "function",
"text": c.Name,
"props": map[string]interface{}{
"id": c.ID,
"name": c.Name,
"arguments": arguments,
},
"function": map[string]interface{}{ "function": map[string]interface{}{
"name": c.Name, "name": c.Name,
"arguments": arguments, "arguments": arguments,