From 3d7b0a2dbd7fa0e929942e33eb030f2bc5ae5335 Mon Sep 17 00:00:00 2001 From: Max Date: Mon, 2 Feb 2026 11:49:10 +0800 Subject: [PATCH] Refactor context ID generation to use UUID - Replace the existing NanoID generation in generateContextID with UUID for improved uniqueness. - Update RequestID method to return the context ID directly instead of generating a new NanoID. Co-authored-by: Cursor --- agent/context/context.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/agent/context/context.go b/agent/context/context.go index 5933318f..75f27c0b 100644 --- a/agent/context/context.go +++ b/agent/context/context.go @@ -5,6 +5,7 @@ import ( "fmt" "sync" + "github.com/google/uuid" "github.com/yaoapp/yao/agent/memory" "github.com/yaoapp/yao/agent/output/message" "github.com/yaoapp/yao/config" @@ -410,12 +411,12 @@ func SendInterrupt(contextID string, signal *InterruptSignal) error { // generateContextID generates a unique context ID func generateContextID() string { - return message.GenerateNanoID() + return uuid.New().String() } // RequestID returns a unique request ID using NanoID func (ctx *Context) RequestID() string { - return message.GenerateNanoID() + return ctx.ID } // TraceID returns the trace ID for the context