phidata: 构建具有记忆、知识和工具的AI助手

phidata

phidata:让AI助手更智能、更自主

在人工智能快速发展的今天,大型语言模型(LLM)的能力令人惊叹。然而,如何将这些通用的LLM转化为能够解决特定问题的专门AI助手,一直是开发者面临的挑战。phidata应运而生,为开发者提供了一个强大的框架,用于构建具有记忆、知识和工具的AI助手。

什么是phidata?

phidata是一个开源框架,旨在帮助开发者构建自主AI助手(也称为Agent)。这些AI助手具有长期记忆、上下文知识,以及通过函数调用执行操作的能力。使用phidata,开发者可以将任何LLM转变为功能强大的AI助手,使其能够:

使用DuckDuckGo、Google等搜索网络
使用SQL、DuckDb等分析数据
进行研究并生成报告
从PDF、API等源回答问题
为电影、书籍等编写剧本
总结文章、视频等内容
执行发送电子邮件、查询数据库等任务

phidata架构图

为什么选择phidata?

phidata的出现解决了一个关键问题:如何将通用的LLM转化为针对特定用例的专门助手。phidata通过为LLM扩展记忆、知识和工具来实现这一目标:

记忆: 在数据库中存储聊天历史,使LLM能够进行长期对话。
知识: 在向量数据库中存储信息,为LLM提供业务上下文。
工具: 使LLM能够执行操作,如从API获取数据、发送电子邮件或查询数据库。

记忆和知识让LLM变得更加智能,而工具则使其具备自主性。这种组合使得基于phidata构建的AI助手能够更好地理解用户需求,并提供更准确、更有价值的响应。

phidata如何工作?

使用phidata构建AI助手的过程可以概括为三个步骤:

创建一个Assistant对象
添加工具(函数)、知识(向量数据库)和存储(数据库)
使用Streamlit、FastAPI或Django来构建和部署AI应用

这种简单而强大的工作流程使得开发者能够快速构建和部署自定义的AI助手,而无需深入了解底层的复杂技术细节。

实际应用示例

为了更好地理解phidata的强大功能,让我们来看几个实际应用的例子:

1. 网络搜索助手

from phi.assistant import Assistant
from phi.tools.duckduckgo import DuckDuckGo

assistant = Assistant(tools=[DuckDuckGo()], show_tool_calls=True)
assistant.print_response("What's happening in France?", markdown=True)

这个简单的例子展示了如何创建一个能够搜索网络的AI助手。通过集成DuckDuckGo工具,助手能够实时获取和处理网络信息,为用户提供最新的答案。

2. 金融数据查询助手

from phi.assistant import Assistant
from phi.llm.openai import OpenAIChat
from phi.tools.yfinance import YFinanceTools

assistant = Assistant(
    llm=OpenAIChat(model="gpt-4o"),
    tools=[YFinanceTools(stock_price=True, analyst_recommendations=True, company_info=True, company_news=True)],
    show_tool_calls=True,
    markdown=True,
)
assistant.print_response("What is the stock price of NVDA")
assistant.print_response("Write a comparison between NVDA and AMD, use all tools available.")

这个例子展示了如何创建一个专门用于金融数据查询的AI助手。通过集成YFinance工具,助手能够获取股票价格、分析师推荐、公司信息和新闻,从而提供深入的金融分析。

3. PDF问答助手

phidata还支持构建能够从PDF文档中回答问题的AI助手。这种助手结合了知识库和存储功能,能够理解PDF内容并保持对话的连贯性。

import typer
from typing import Optional, List
from phi.assistant import Assistant
from phi.storage.assistant.postgres import PgAssistantStorage
from phi.knowledge.pdf import PDFUrlKnowledgeBase
from phi.vectordb.pgvector import PgVector2

# 配置数据库连接
db_url = "postgresql+psycopg://ai:ai@localhost:5532/ai"

# 创建知识库
knowledge_base = PDFUrlKnowledgeBase(
    urls=["https://phi-public.s3.amazonaws.com/recipes/ThaiRecipes.pdf"],
    vector_db=PgVector2(collection="recipes", db_url=db_url),
)
# 首次运行时加载知识库
knowledge_base.load()

# 配置存储
storage = PgAssistantStorage(table_name="pdf_assistant", db_url=db_url)

# 创建PDF助手
def pdf_assistant(new: bool = False, user: str = "user"):
    run_id: Optional[str] = None

    if not new:
        existing_run_ids: List[str] = storage.get_all_run_ids(user)
        if len(existing_run_ids) > 0:
            run_id = existing_run_ids[0]

    assistant = Assistant(
        run_id=run_id,
        user_id=user,
        knowledge_base=knowledge_base,
        storage=storage,
        show_tool_calls=True,
        search_knowledge=True,
        read_chat_history=True,
    )
    
    # 运行助手作为CLI应用
    assistant.cli_app(markdown=True)

if __name__ == "__main__":
    typer.run(pdf_assistant)

这个PDF助手能够从指定的PDF文档中提取信息,回答用户的问题,并在多个会话之间保持对话的连贯性。

phidata的优势

灵活性: phidata支持多种LLM,包括OpenAI的GPT模型、Anthropic的Claude等,开发者可以根据需求选择合适的模型。
可扩展性: 通过添加自定义工具和知识库,开发者可以轻松扩展AI助手的功能。
长期记忆: 内置的存储功能使AI助手能够在多个会话之间保持上下文,提供更连贯的用户体验。
知识整合: 向量数据库的集成使AI助手能够快速检索和利用大量结构化和非结构化数据。
开源生态: 作为开源项目,phidata拥有活跃的社区支持,不断有新的功能和改进被添加。

结语

phidata为AI开发者提供了一个强大而灵活的框架,使构建智能、自主的AI助手变得前所未有的简单。无论是构建简单的聊天机器人,还是复杂的数据分析工具,phidata都能满足各种需求。随着AI技术的不断发展,phidata将继续扮演重要角色,帮助开发者将创新的AI应用带入现实世界。

对于那些希望深入了解phidata或开始构建自己的AI助手的开发者,phidata的官方文档和GitHub仓库提供了丰富的资源和示例。现在正是探索和利用这一强大工具的最佳时机,让我们一起推动AI应用的边界,创造更智能、更有价值的解决方案。