Fix/memory bug fix (#171)
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
import uuid
|
||||
from uuid import UUID
|
||||
|
||||
from pydantic import Field
|
||||
from typing import Literal
|
||||
@@ -11,7 +12,7 @@ class MemoryReadNodeConfig(BaseNodeConfig):
|
||||
...
|
||||
)
|
||||
|
||||
config_id: int = Field(
|
||||
config_id: UUID = Field(
|
||||
...
|
||||
)
|
||||
|
||||
@@ -26,6 +27,6 @@ class MemoryWriteNodeConfig(BaseNodeConfig):
|
||||
...
|
||||
)
|
||||
|
||||
config_id: int = Field(
|
||||
config_id: UUID = Field(
|
||||
...
|
||||
)
|
||||
|
||||
@@ -22,7 +22,7 @@ class MemoryReadNode(BaseNode):
|
||||
raise RuntimeError("End user id is required")
|
||||
|
||||
return await MemoryAgentService().read_memory(
|
||||
group_id=end_user_id,
|
||||
end_user_id=end_user_id,
|
||||
message=self._render_template(self.typed_config.message, state),
|
||||
config_id=str(self.typed_config.config_id),
|
||||
search_switch=self.typed_config.search_switch,
|
||||
|
||||
Reference in New Issue
Block a user