agentica 0.0.1__tar.gz → 0.1.0__tar.gz

agentica-0.1.0/PKG-INFO

@@ -0,0 +1,199 @@
+Metadata-Version: 2.1
+Name: agentica
+Version: 0.1.0
+Summary: LLM agents
+Home-page: https://github.com/shibing624/agentica
+Author: XuMing
+Author-email: xuming624@qq.com
+License: Apache License 2.0
+Keywords: Agentica,Agent Tool,action,agent,agentica
+Platform: UNKNOWN
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Education
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.8.0
+Description-Content-Type: text/markdown
+License-File: LICENSE
+
+[**🇨🇳中文**](https://github.com/shibing624/agentica/blob/main/README.md) | [**🌐English**](https://github.com/shibing624/agentica/blob/main/README_EN.md)
+
+<div align="center">
+  <a href="https://github.com/shibing624/agentica">
+    <img src="https://raw.githubusercontent.com/shibing624/agentica/main/docs/logo.png" height="150" alt="Logo">
+  </a>
+</div>
+
+-----------------
+
+# Agentica: Build AI Agents
+[![PyPI version](https://badge.fury.io/py/agentica.svg)](https://badge.fury.io/py/agentica)
+[![Downloads](https://static.pepy.tech/badge/agentica)](https://pepy.tech/project/agentica)
+[![Contributions welcome](https://img.shields.io/badge/contributions-welcome-brightgreen.svg)](CONTRIBUTING.md)
+[![License Apache 2.0](https://img.shields.io/badge/license-Apache%202.0-blue.svg)](LICENSE)
+[![python_version](https://img.shields.io/badge/Python-3.5%2B-green.svg)](requirements.txt)
+[![GitHub issues](https://img.shields.io/github/issues/shibing624/agentica.svg)](https://github.com/shibing624/agentica/issues)
+[![Wechat Group](https://img.shields.io/badge/wechat-group-green.svg?logo=wechat)](#Contact)
+
+
+**agentica**: A Human-Centric Framework for Large Language Model Agent Building.
+
+**agentica**: 构建你自己的Agent
+
+## Overview
+
+#### LLM Agent
+![llm_agnet](https://github.com/shibing624/agentica/blob/main/docs/llm_agent.png)
+
+- **规划（Planning）**：任务拆解、生成计划、反思
+- **记忆（Memory）**：短期记忆（prompt实现）、长期记忆（RAG实现）
+- **工具使用（Tool use）**：function call能力，调用外部API，以获取外部信息，包括当前日期、日历、代码执行能力、对专用信息源的访问等
+
+#### agentica架构
+![agentica_arch](https://github.com/shibing624/agentica/blob/main/docs/agent_arch.png)
+
+- **Planner**：负责让LLM生成一个多步计划来完成复杂任务，生成相互依赖的“链式计划”，定义每一步所依赖的上一步的输出
+- **Worker**：接受“链式计划”，循环遍历计划中的每个子任务，并调用工具完成任务，可以自动反思纠错以完成任务
+- **Solver**：求解器将所有这些输出整合为最终答案
+
+## Features
+`agentica`是一个Agent构建工具，功能：
+
+- 简单代码快速编排Agent，支持 Reflection(反思）、Plan and Solve(计划并执行)、RAG、Agent、Multi-Agent、Multi-Role、Workflow等功能
+- Agent支持prompt自定义，支持多种工具调用（tool_calls）
+- 支持OpenAI/Azure/Claude/Ollama/Together API调用
+
+## Installation
+
+```bash
+pip install -U agentica
+```
+
+or
+
+```bash
+git clone https://github.com/shibing624/agentica.git
+cd agentica
+pip install .
+```
+
+## Getting Started
+
+1. 复制[example.env](https://github.com/shibing624/agentica/blob/main/example.env)文件为`.env`，并粘贴OpenAI API key或者Moonshoot API key。
+
+2. 使用`agentica`构建Agent，拆解任务并执行：
+
+自动调用google搜索工具，示例[examples/web_search_demo.py](https://github.com/shibing624/agentica/blob/main/examples/web_search_demo.py)
+
+```python
+from agentica import Assistant, OpenAILLM, AzureOpenAILLM
+from agentica.tools.search_serper import SearchSerperTool
+
+m = Assistant(
+  llm=AzureOpenAILLM(),
+  description="You are a helpful ai assistant.",
+  show_tool_calls=True,
+  # Enable the assistant to search the knowledge base
+  search_knowledge=False,
+  tools=[SearchSerperTool()],
+  # Enable the assistant to read the chat history
+  read_chat_history=True,
+  debug_mode=True,
+)
+
+r = m.run("一句话介绍林黛玉")
+print(r, "".join(r))
+r = m.run("北京最近的新闻", stream=True, print_output=True)
+print(r, "".join(r))
+r = m.run("总结前面的问答", stream=False, print_output=False)
+print(r)
+```
+
+
+## Examples
+
+| 示例                                                                                                                                    | 描述                                                                                                                              |
+|---------------------------------------------------------------------------------------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------|
+| [examples/naive_rag_demo.py](https://github.com/shibing624/agentica/blob/main/examples/naive_rag_demo.py)                             | 实现了基础版RAG，基于Txt文档回答问题                                                                                                           |
+| [examples/advanced_rag_demo.py](https://github.com/shibing624/agentica/blob/main/examples/advanced_rag_demo.py)                       | 实现了高级版RAG，基于PDF文档回答问题，新增功能：pdf文件解析、query改写，字面+语义多路召回，召回排序（rerank）                                                               |
+| [examples/python_assistant_demo.py](https://github.com/shibing624/agentica/blob/main/examples/python_assistant_demo.py)               | 实现了Code Interpreter功能，自动生成python代码，并执行                                                                                          |
+| [examples/research_demo.py](https://github.com/shibing624/agentica/blob/main/examples/research_demo.py)                               | 实现了Research功能，自动调用搜索工具，汇总信息后撰写科技报告                                                                                              |
+| [examples/team_news_article_demo.py](https://github.com/shibing624/agentica/blob/main/examples/team_news_article_demo.py)             | 实现了写新闻稿的team协作，multi-role实现，委托不用角色完成各自任务：研究员检索分析文章，撰写员根据排版写文章，汇总多角色成果输出结果                                                       |
+| [examples/workflow_news_article_demo.py](https://github.com/shibing624/agentica/blob/main/examples/workflow_news_article_demo.py)     | 实现了写新闻稿的工作流，multi-agent的实现，定义了多个Assistant和Task，多次调用搜索工具，并生成高级排版的新闻文章                                                            |
+| [examples/workflow_investment_demo.py](https://github.com/shibing624/agentica/blob/main/examples/workflow_investment_demo.py)         | 实现了投资研究的工作流：股票信息收集 - 股票分析 - 撰写分析报告 - 复查报告等多个Task                                                                                |
+| [examples/crawl_webpage_demo.py](https://github.com/shibing624/agentica/blob/main/examples/crawl_webpage_demo.py)                          | 实现了网页分析工作流：从Url爬取融资快讯 - 分析网页内容和格式 - 提取核心信息 - 汇总保存为md文件                                                                          |
+| [examples/find_paper_from_arxiv_demo.py](https://github.com/shibing624/agentica/blob/main/examples/find_paper_from_arxiv_demo.py)     | 实现了论文推荐工作流：自动从arxiv搜索多组论文 - 相似论文去重 - 提取核心论文信息 - 保存为csv文件                                                                        |
+| [examples/remove_image_background_demo.py](https://github.com/shibing624/agentica/blob/main/examples/remove_image_background_demo.py) | 实现了自动去除图片背景功能，包括自动通过pip安装库，调用库实现去除图片背景                                                                                          |
+| [examples/text_classification_demo.py](https://github.com/shibing624/agentica/blob/main/examples/text_classification_demo.py)         | 实现了自动训练分类模型的工作流：读取训练集文件并理解格式 - 谷歌搜索pytextclassifier库 - 爬取github页面了解pytextclassifier的调用方法 - 写代码并执行fasttext模型训练 - check训练好的模型预测结果 |
+| [examples/llm_os_demo.py](https://github.com/shibing624/agentica/blob/main/examples/llm_os_demo.py)                                   | 实现了LLM OS的初步设计，基于LLM设计操作系统，可以通过LLM调用RAG、代码执行器、Shell等工具，并协同代码解释器、研究助手、投资助手等来解决问题。                                                |
+
+
+### LLM OS
+The LLM OS design:
+<img alt="LLM OS" src="https://github.com/shibing624/agentica/blob/main/docs/llmos.png" width="600" />
+
+#### Run the LLM OS App
+
+```shell
+cd examples
+streamlit run llm_os_demo.py
+```
+
+![llm_os](https://github.com/shibing624/agentica/blob/main/docs/llm_os_snap.png)
+
+## Contact
+
+- Issue(建议)
+  ：[![GitHub issues](https://img.shields.io/github/issues/shibing624/agentica.svg)](https://github.com/shibing624/agentica/issues)
+- 邮件我：xuming: xuming624@qq.com
+- 微信我： 加我*微信号：xuming624, 备注：姓名-公司-NLP* 进NLP交流群。
+
+<img src="https://github.com/shibing624/agentica/blob/main/docs/wechat.jpeg" width="200" />
+
+## Citation
+
+如果你在研究中使用了`agentica`，请按如下格式引用：
+
+APA:
+
+```
+Xu, M. agentica: A Human-Centric Framework for Large Language Model Agent Workflows (Version 0.0.2) [Computer software]. https://github.com/shibing624/agentica
+```
+
+BibTeX:
+
+```
+@misc{Xu_agentica,
+  title={agentica: A Human-Centric Framework for Large Language Model Agent Workflows},
+  author={Xu Ming},
+  year={2024},
+  howpublished={\url{https://github.com/shibing624/agentica}},
+}
+```
+
+## License
+
+授权协议为 [The Apache License 2.0](/LICENSE)，可免费用做商业用途。请在产品说明中附加`agentica`的链接和授权协议。
+## Contribute
+
+项目代码还很粗糙，如果大家对代码有所改进，欢迎提交回本项目，在提交之前，注意以下两点：
+
+- 在`tests`添加相应的单元测试
+- 使用`python -m pytest`来运行所有单元测试，确保所有单测都是通过的
+
+之后即可提交PR。
+
+## Acknowledgements 
+
+- [https://github.com/langchain-ai/langchain](https://github.com/langchain-ai/langchain)
+- [https://github.com/simonmesmith/agentflow](https://github.com/simonmesmith/agentflow)
+- [https://github.com/phidatahq/phidata](https://github.com/phidatahq/phidata)
+
+
+Thanks for their great work!
+
+

agentica-0.1.0/README.md

@@ -0,0 +1,175 @@
+[**🇨🇳中文**](https://github.com/shibing624/agentica/blob/main/README.md) | [**🌐English**](https://github.com/shibing624/agentica/blob/main/README_EN.md)
+
+<div align="center">
+  <a href="https://github.com/shibing624/agentica">
+    <img src="https://raw.githubusercontent.com/shibing624/agentica/main/docs/logo.png" height="150" alt="Logo">
+  </a>
+</div>
+
+-----------------
+
+# Agentica: Build AI Agents
+[![PyPI version](https://badge.fury.io/py/agentica.svg)](https://badge.fury.io/py/agentica)
+[![Downloads](https://static.pepy.tech/badge/agentica)](https://pepy.tech/project/agentica)
+[![Contributions welcome](https://img.shields.io/badge/contributions-welcome-brightgreen.svg)](CONTRIBUTING.md)
+[![License Apache 2.0](https://img.shields.io/badge/license-Apache%202.0-blue.svg)](LICENSE)
+[![python_version](https://img.shields.io/badge/Python-3.5%2B-green.svg)](requirements.txt)
+[![GitHub issues](https://img.shields.io/github/issues/shibing624/agentica.svg)](https://github.com/shibing624/agentica/issues)
+[![Wechat Group](https://img.shields.io/badge/wechat-group-green.svg?logo=wechat)](#Contact)
+
+
+**agentica**: A Human-Centric Framework for Large Language Model Agent Building.
+
+**agentica**: 构建你自己的Agent
+
+## Overview
+
+#### LLM Agent
+![llm_agnet](https://github.com/shibing624/agentica/blob/main/docs/llm_agent.png)
+
+- **规划（Planning）**：任务拆解、生成计划、反思
+- **记忆（Memory）**：短期记忆（prompt实现）、长期记忆（RAG实现）
+- **工具使用（Tool use）**：function call能力，调用外部API，以获取外部信息，包括当前日期、日历、代码执行能力、对专用信息源的访问等
+
+#### agentica架构
+![agentica_arch](https://github.com/shibing624/agentica/blob/main/docs/agent_arch.png)
+
+- **Planner**：负责让LLM生成一个多步计划来完成复杂任务，生成相互依赖的“链式计划”，定义每一步所依赖的上一步的输出
+- **Worker**：接受“链式计划”，循环遍历计划中的每个子任务，并调用工具完成任务，可以自动反思纠错以完成任务
+- **Solver**：求解器将所有这些输出整合为最终答案
+
+## Features
+`agentica`是一个Agent构建工具，功能：
+
+- 简单代码快速编排Agent，支持 Reflection(反思）、Plan and Solve(计划并执行)、RAG、Agent、Multi-Agent、Multi-Role、Workflow等功能
+- Agent支持prompt自定义，支持多种工具调用（tool_calls）
+- 支持OpenAI/Azure/Claude/Ollama/Together API调用
+
+## Installation
+
+```bash
+pip install -U agentica
+```
+
+or
+
+```bash
+git clone https://github.com/shibing624/agentica.git
+cd agentica
+pip install .
+```
+
+## Getting Started
+
+1. 复制[example.env](https://github.com/shibing624/agentica/blob/main/example.env)文件为`.env`，并粘贴OpenAI API key或者Moonshoot API key。
+
+2. 使用`agentica`构建Agent，拆解任务并执行：
+
+自动调用google搜索工具，示例[examples/web_search_demo.py](https://github.com/shibing624/agentica/blob/main/examples/web_search_demo.py)
+
+```python
+from agentica import Assistant, OpenAILLM, AzureOpenAILLM
+from agentica.tools.search_serper import SearchSerperTool
+
+m = Assistant(
+  llm=AzureOpenAILLM(),
+  description="You are a helpful ai assistant.",
+  show_tool_calls=True,
+  # Enable the assistant to search the knowledge base
+  search_knowledge=False,
+  tools=[SearchSerperTool()],
+  # Enable the assistant to read the chat history
+  read_chat_history=True,
+  debug_mode=True,
+)
+
+r = m.run("一句话介绍林黛玉")
+print(r, "".join(r))
+r = m.run("北京最近的新闻", stream=True, print_output=True)
+print(r, "".join(r))
+r = m.run("总结前面的问答", stream=False, print_output=False)
+print(r)
+```
+
+
+## Examples
+
+| 示例                                                                                                                                    | 描述                                                                                                                              |
+|---------------------------------------------------------------------------------------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------|
+| [examples/naive_rag_demo.py](https://github.com/shibing624/agentica/blob/main/examples/naive_rag_demo.py)                             | 实现了基础版RAG，基于Txt文档回答问题                                                                                                           |
+| [examples/advanced_rag_demo.py](https://github.com/shibing624/agentica/blob/main/examples/advanced_rag_demo.py)                       | 实现了高级版RAG，基于PDF文档回答问题，新增功能：pdf文件解析、query改写，字面+语义多路召回，召回排序（rerank）                                                               |
+| [examples/python_assistant_demo.py](https://github.com/shibing624/agentica/blob/main/examples/python_assistant_demo.py)               | 实现了Code Interpreter功能，自动生成python代码，并执行                                                                                          |
+| [examples/research_demo.py](https://github.com/shibing624/agentica/blob/main/examples/research_demo.py)                               | 实现了Research功能，自动调用搜索工具，汇总信息后撰写科技报告                                                                                              |
+| [examples/team_news_article_demo.py](https://github.com/shibing624/agentica/blob/main/examples/team_news_article_demo.py)             | 实现了写新闻稿的team协作，multi-role实现，委托不用角色完成各自任务：研究员检索分析文章，撰写员根据排版写文章，汇总多角色成果输出结果                                                       |
+| [examples/workflow_news_article_demo.py](https://github.com/shibing624/agentica/blob/main/examples/workflow_news_article_demo.py)     | 实现了写新闻稿的工作流，multi-agent的实现，定义了多个Assistant和Task，多次调用搜索工具，并生成高级排版的新闻文章                                                            |
+| [examples/workflow_investment_demo.py](https://github.com/shibing624/agentica/blob/main/examples/workflow_investment_demo.py)         | 实现了投资研究的工作流：股票信息收集 - 股票分析 - 撰写分析报告 - 复查报告等多个Task                                                                                |
+| [examples/crawl_webpage_demo.py](https://github.com/shibing624/agentica/blob/main/examples/crawl_webpage_demo.py)                          | 实现了网页分析工作流：从Url爬取融资快讯 - 分析网页内容和格式 - 提取核心信息 - 汇总保存为md文件                                                                          |
+| [examples/find_paper_from_arxiv_demo.py](https://github.com/shibing624/agentica/blob/main/examples/find_paper_from_arxiv_demo.py)     | 实现了论文推荐工作流：自动从arxiv搜索多组论文 - 相似论文去重 - 提取核心论文信息 - 保存为csv文件                                                                        |
+| [examples/remove_image_background_demo.py](https://github.com/shibing624/agentica/blob/main/examples/remove_image_background_demo.py) | 实现了自动去除图片背景功能，包括自动通过pip安装库，调用库实现去除图片背景                                                                                          |
+| [examples/text_classification_demo.py](https://github.com/shibing624/agentica/blob/main/examples/text_classification_demo.py)         | 实现了自动训练分类模型的工作流：读取训练集文件并理解格式 - 谷歌搜索pytextclassifier库 - 爬取github页面了解pytextclassifier的调用方法 - 写代码并执行fasttext模型训练 - check训练好的模型预测结果 |
+| [examples/llm_os_demo.py](https://github.com/shibing624/agentica/blob/main/examples/llm_os_demo.py)                                   | 实现了LLM OS的初步设计，基于LLM设计操作系统，可以通过LLM调用RAG、代码执行器、Shell等工具，并协同代码解释器、研究助手、投资助手等来解决问题。                                                |
+
+
+### LLM OS
+The LLM OS design:
+<img alt="LLM OS" src="https://github.com/shibing624/agentica/blob/main/docs/llmos.png" width="600" />
+
+#### Run the LLM OS App
+
+```shell
+cd examples
+streamlit run llm_os_demo.py
+```
+
+![llm_os](https://github.com/shibing624/agentica/blob/main/docs/llm_os_snap.png)
+
+## Contact
+
+- Issue(建议)
+  ：[![GitHub issues](https://img.shields.io/github/issues/shibing624/agentica.svg)](https://github.com/shibing624/agentica/issues)
+- 邮件我：xuming: xuming624@qq.com
+- 微信我： 加我*微信号：xuming624, 备注：姓名-公司-NLP* 进NLP交流群。
+
+<img src="https://github.com/shibing624/agentica/blob/main/docs/wechat.jpeg" width="200" />
+
+## Citation
+
+如果你在研究中使用了`agentica`，请按如下格式引用：
+
+APA:
+
+```
+Xu, M. agentica: A Human-Centric Framework for Large Language Model Agent Workflows (Version 0.0.2) [Computer software]. https://github.com/shibing624/agentica
+```
+
+BibTeX:
+
+```
+@misc{Xu_agentica,
+  title={agentica: A Human-Centric Framework for Large Language Model Agent Workflows},
+  author={Xu Ming},
+  year={2024},
+  howpublished={\url{https://github.com/shibing624/agentica}},
+}
+```
+
+## License
+
+授权协议为 [The Apache License 2.0](/LICENSE)，可免费用做商业用途。请在产品说明中附加`agentica`的链接和授权协议。
+## Contribute
+
+项目代码还很粗糙，如果大家对代码有所改进，欢迎提交回本项目，在提交之前，注意以下两点：
+
+- 在`tests`添加相应的单元测试
+- 使用`python -m pytest`来运行所有单元测试，确保所有单测都是通过的
+
+之后即可提交PR。
+
+## Acknowledgements 
+
+- [https://github.com/langchain-ai/langchain](https://github.com/langchain-ai/langchain)
+- [https://github.com/simonmesmith/agentflow](https://github.com/simonmesmith/agentflow)
+- [https://github.com/phidatahq/phidata](https://github.com/phidatahq/phidata)
+
+
+Thanks for their great work!

agentica-0.1.0/agentica/__init__.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+"""
+@author:XuMing(xuming624@qq.com)
+@description: 
+"""
+from agentica.version import __version__  # noqa, isort: skip
+from agentica.config import DOTENV_PATH, SMART_LLM, FAST_LLM  # noqa, isort: skip
+from agentica.assistant import Assistant
+from agentica.python_assistant import PythonAssistant
+from agentica.document import Document
+from agentica.knowledge_base import KnowledgeBase
+from agentica.llm.openai_llm import OpenAILLM
+from agentica.llm.azure_llm import AzureOpenAILLM
+from agentica.task import Task
+from agentica.workflow import Workflow
+

{agentica-0.0.1/actionflow → agentica-0.1.0/agentica}/assistant.py

@@ -28,18 +28,18 @@ from uuid import uuid4
 
 from pydantic import BaseModel, ConfigDict, field_validator, ValidationError
 
-from actionflow.document import Document
-from actionflow.documents import Documents
-from actionflow.llm.base import LLM
-from actionflow.memory import AssistantMemory, Memory
-from actionflow.message import Message
-from actionflow.references import References
-from actionflow.run_record import RunRecord
-from actionflow.sqlite_storage import SqliteStorage
-from actionflow.tool import Tool, Toolkit, Function
-from actionflow.utils.log import logger, set_log_level_to_debug, print_llm_stream
-from actionflow.utils.misc import merge_dictionaries, remove_indent
-from actionflow.utils.timer import Timer
+from agentica.document import Document
+from agentica.knowledge_base import KnowledgeBase
+from agentica.llm.base import LLM
+from agentica.memory import AssistantMemory, Memory
+from agentica.message import Message
+from agentica.references import References
+from agentica.run_record import RunRecord
+from agentica.sqlite_storage import SqliteStorage
+from agentica.tool import Tool, Toolkit, Function
+from agentica.utils.log import logger, set_log_level_to_debug, print_llm_stream
+from agentica.utils.misc import merge_dictionaries, remove_indent
+from agentica.utils.timer import Timer
 
 
 class Assistant(BaseModel):
@@ -81,7 +81,7 @@ class Assistant(BaseModel):
     update_memory_after_run: bool = True
 
     # -*- Assistant Knowledge Base
-    knowledge_base: Optional[Documents] = None
+    knowledge_base: Optional[KnowledgeBase] = None
     # Enable RAG by adding references from the knowledge base to the prompt.
     add_references_to_prompt: bool = False
 
@@ -269,7 +269,7 @@ class Assistant(BaseModel):
     def update_llm(self) -> None:
         if self.llm is None:
             try:
-                from actionflow.llm.openai_llm import OpenAILLM
+                from agentica.llm.openai_llm import OpenAILLM
             except ModuleNotFoundError as e:
                 logger.exception(e)
                 logger.error("use `openai` as the default LLM. ")
@@ -790,6 +790,7 @@ class Assistant(BaseModel):
             *,
             stream: bool = True,
             messages: Optional[List[Union[Dict, Message]]] = None,
+            print_output: bool = True,
             **kwargs: Any,
     ) -> Iterator[str]:
         logger.debug(f"*********** Assistant Run Start: {self.run_id} ***********")
@@ -865,6 +866,8 @@ class Assistant(BaseModel):
         self.llm = cast(LLM, self.llm)
         if stream and self.streamable:
             for response_chunk in self.llm.response_stream(messages=llm_messages):
+                if print_output:
+                    print_llm_stream(response_chunk)
                 llm_response += response_chunk
                 yield response_chunk
         else:
@@ -922,6 +925,8 @@ class Assistant(BaseModel):
 
         # -*- Yield final response if not streaming
         if not stream:
+            if print_output:
+                print(llm_response)
             yield llm_response
 
     def run(
@@ -929,8 +934,8 @@ class Assistant(BaseModel):
             message: Optional[Union[List, Dict, str]] = None,
             *,
             stream: bool = True,
-            print_output: bool = True,
             messages: Optional[List[Union[Dict, Message]]] = None,
+            print_output: bool = True,
             **kwargs: Any,
     ) -> Union[Iterator[str], str, BaseModel]:
         # Convert response to structured output if output_model is set
@@ -959,16 +964,11 @@ class Assistant(BaseModel):
             return self.output or json_resp
         else:
             if stream and self.streamable:
-                resp = self._run(message=message, messages=messages, stream=True, **kwargs)
-                if print_output:
-                    for chunk in resp:
-                        print_llm_stream(chunk)
+                resp = self._run(message=message, messages=messages, stream=True, print_output=print_output, **kwargs)
                 return resp
             else:
-                resp = self._run(message=message, messages=messages, stream=False, **kwargs)
+                resp = self._run(message=message, messages=messages, stream=False, print_output=print_output, **kwargs)
                 resp = next(resp)
-                if print_output:
-                    print(resp)
                 return resp
 
     async def _arun(
@@ -977,6 +977,7 @@ class Assistant(BaseModel):
             *,
             stream: bool = True,
             messages: Optional[List[Union[Dict, Message]]] = None,
+            print_output: bool = True,
             **kwargs: Any,
     ) -> AsyncIterator[str]:
         logger.debug(f"*********** Run Start: {self.run_id} ***********")
@@ -1055,6 +1056,8 @@ class Assistant(BaseModel):
             response_stream = self.llm.aresponse_stream(messages=llm_messages)
             async for response_chunk in response_stream:  # type: ignore
                 llm_response += response_chunk
+                if print_output:
+                    print_llm_stream(response_chunk)
                 yield response_chunk
         else:
             llm_response = await self.llm.aresponse(messages=llm_messages)
@@ -1089,6 +1092,8 @@ class Assistant(BaseModel):
 
         # -*- Yield final response if not streaming
         if not stream:
+            if print_output:
+                print(llm_response)
             yield llm_response
 
     async def arun(
@@ -1096,8 +1101,8 @@ class Assistant(BaseModel):
             message: Optional[Union[List, Dict, str]] = None,
             *,
             stream: bool = True,
-            print_output: bool = True,
             messages: Optional[List[Union[Dict, Message]]] = None,
+            print_output: bool = True,
             **kwargs: Any,
     ) -> Union[AsyncIterator[str], str, BaseModel]:
         # Convert response to structured output if output_model is set
@@ -1127,17 +1132,11 @@ class Assistant(BaseModel):
             return self.output or json_resp
         else:
             if stream and self.streamable:
-                resp = self._arun(message=message, messages=messages, stream=True, **kwargs)
-                if print_output:
-                    async for chunk in resp:
-                        print_llm_stream(chunk)
+                resp = self._arun(message=message, messages=messages, stream=True, print_output=print_output, **kwargs)
                 return resp
             else:
-                resp = self._arun(message=message, messages=messages, stream=False, **kwargs)
-                r = await resp.__anext__()
-                if print_output:
-                    print(r)
-                return r
+                resp = self._arun(message=message, messages=messages, stream=False, print_output=print_output, **kwargs)
+                return await resp.__anext__()
 
     def chat(
             self, message: Union[List, Dict, str], stream: bool = True, **kwargs: Any
@@ -1334,7 +1333,6 @@ class Assistant(BaseModel):
             user: str = "User",
             emoji: str = ":sunglasses:",
             stream: bool = True,
-            markdown: bool = False,
             exit_on: Optional[List[str]] = None,
             **kwargs: Any,
     ) -> None:
@@ -1342,11 +1340,11 @@ class Assistant(BaseModel):
         _exit_on = exit_on or ["exit", "quit", "bye"]
         logger.debug(f"Enable cli, exit with {_exit_on[0]}")
         if message:
-            self.run(message=message, stream=stream, markdown=markdown, **kwargs)
+            self.run(message=message, stream=stream, **kwargs)
 
         while True:
             message = Prompt.ask(f"[bold] {emoji} {user} [/bold]")
             if message in _exit_on:
                 break
 
-            self.run(message=message, stream=stream, markdown=markdown, **kwargs)
+            self.run(message=message, stream=stream, **kwargs)

{agentica-0.0.1/actionflow → agentica-0.1.0/agentica}/config.py

@@ -19,10 +19,10 @@ try:
 except ImportError:
     logger.debug("dotenv not installed, skipping...")
 
-ACTIONFLOW_HOME = os.environ.get("ACTIONFLOW_HOME", os.path.realpath(os.path.curdir))
+AGENTICA_HOME = os.environ.get("AGENTICA_HOME", os.path.realpath(os.path.curdir))
 current_date = datetime.now()
 formatted_date = current_date.strftime("%Y%m%d")
-LOG_FILE = os.environ.get("LOG_FILE", f"{ACTIONFLOW_HOME}/logs/{formatted_date}.log")
+LOG_FILE = os.environ.get("LOG_FILE", f"{AGENTICA_HOME}/logs/{formatted_date}.log")
 LOG_LEVEL = os.environ.get("LOG_LEVEL", "INFO")
 SMART_LLM = os.environ.get("SMART_LLM")
 FAST_LLM = os.environ.get("FAST_LLM")

{agentica-0.0.1/actionflow → agentica-0.1.0/agentica}/document.py

@@ -9,7 +9,7 @@ from typing import Optional, Dict, Any, List
 
 from pydantic import BaseModel, ConfigDict
 
-from actionflow.emb.base import Emb
+from agentica.emb.base import Emb
 
 
 class Document(BaseModel):

{agentica-0.0.1/actionflow → agentica-0.1.0/agentica}/emb/azure_emb.py

@@ -8,8 +8,8 @@ from os import getenv
 from typing import Optional, Dict, List, Tuple, Any
 from typing_extensions import Literal
 
-from actionflow.emb.base import Emb
-from actionflow.utils.log import logger
+from agentica.emb.base import Emb
+from agentica.utils.log import logger
 
 try:
     from openai import AzureOpenAI as AzureOpenAIClient

{agentica-0.0.1/actionflow → agentica-0.1.0/agentica}/emb/hash_emb.py

@@ -6,7 +6,7 @@
 import hashlib
 from typing import List
 
-from actionflow.emb.base import Emb
+from agentica.emb.base import Emb
 
 
 class HashEmb(Emb):

{agentica-0.0.1/actionflow → agentica-0.1.0/agentica}/emb/ollama_emb.py

@@ -6,8 +6,8 @@ part of the code from https://github.com/phidatahq/phidata
 """
 from typing import Optional, Dict, List, Any
 
-from actionflow.emb.base import Emb
-from actionflow.utils.log import logger
+from agentica.emb.base import Emb
+from agentica.utils.log import logger
 
 try:
     from ollama import Client as OllamaClient

{agentica-0.0.1/actionflow → agentica-0.1.0/agentica}/emb/openai_emb.py

@@ -9,8 +9,8 @@ from typing import Optional, Dict, List, Tuple, Any
 
 from typing_extensions import Literal
 
-from actionflow.emb.base import Emb
-from actionflow.utils.log import logger
+from agentica.emb.base import Emb
+from agentica.utils.log import logger
 
 try:
     from openai import OpenAI as OpenAIClient

{agentica-0.0.1/actionflow → agentica-0.1.0/agentica}/emb/text2vec_emb.py

@@ -11,7 +11,7 @@ except ImportError:
     raise ImportError(
         "`text2vec` not installed. Please install it with `pip install text2vec`"
     )
-from actionflow.emb.base import Emb
+from agentica.emb.base import Emb
 
 
 class Text2VecEmb(Emb):