Merge branch 'feature/merge/v0.6-release' into geekan/dev

This commit is contained in:
莘权 马 2024-01-21 16:00:35 +08:00
commit 0f8c203eff
4 changed files with 16 additions and 6 deletions

View file

@ -37,7 +37,11 @@ class FaissStore(LocalStore):
return FAISS.load_local(self.raw_data_path.parent, self.embedding, self.fname)
def _write(self, docs, metadatas):
store = FAISS.from_texts(docs, self.embedding, metadatas=metadatas)
try:
store = FAISS.from_texts(docs, self.embedding, metadatas=metadatas)
except Exception as e:
logger.error(f"Failed to write. error: {e}")
raise e
return store
def persist(self):

View file

@ -297,6 +297,6 @@ class Engineer(Role):
self.set_todo(self.summarize_todos[0])
@property
def todo(self) -> str:
def action_description(self) -> str:
"""AgentStore uses this attribute to display to the user what actions the current role should take."""
return self.next_todo_action

View file

@ -524,6 +524,7 @@ class Role(SerializationMixin, ContextMixin, BaseModel):
async def think(self) -> Action:
"""The exported `think` function"""
await self._observe()
await self._think()
return self.rc.todo
@ -533,9 +534,14 @@ class Role(SerializationMixin, ContextMixin, BaseModel):
return ActionOutput(content=msg.content, instruct_content=msg.instruct_content)
@property
def first_action(self) -> str:
"""AgentStore uses this attribute to display to the user what actions the current role should take."""
# FIXME: this is a hack, we should not use the first action to represent the todo
def action_description(self) -> str:
"""
AgentStore uses this attribute to display to the user what actions the current role should take.
"""
if self.rc.todo:
if self.rc.todo.desc:
return self.rc.todo.desc
return any_to_name(self.rc.todo)
if self.actions:
return any_to_name(self.actions[0])
return ""

View file

@ -14,7 +14,7 @@ lancedb==0.4.0
langchain==0.0.352
loguru==0.6.0
meilisearch==0.21.0
numpy==1.24.3
numpy>=1.24.3
openai==1.6.0
openpyxl
beautifulsoup4==4.12.2