github

AI4Finance-Foundation / FinRobot

  • пятница, 31 мая 2024β€―Π³. Π² 00:00:07
https://github.com/AI4Finance-Foundation/FinRobot

FinRobot: An Open-Source AI Agent Platform for Financial Applications using LLMs πŸš€ πŸš€ πŸš€



image

FinRobot: An Open-Source AI Agent Platform for Financial Applications using Large Language Models

Downloads Downloads Python 3.8 PyPI License

FinRobot is an AI Agent Platform that transcends the scope of FinGPT, representing a comprehensive solution meticulously designed for financial applications. It integrates a diverse array of AI technologies, extending beyond mere language models. This expansive vision highlights the platform's versatility and adaptability, addressing the multifaceted needs of the financial industry.

Concept of AI Agent: an AI Agent is an intelligent entity that uses large language models as its brain to perceive its environment, make decisions, and execute actions. Unlike traditional artificial intelligence, AI Agents possess the ability to independently think and utilize tools to progressively achieve given objectives.

Whitepaper of FinRobot

FinRobot Ecosystem

The overall framework of FinRobot is organized into four distinct layers, each designed to address specific aspects of financial AI processing and application:

  1. Financial AI Agents Layer: The Financial AI Agents Layer now includes Financial Chain-of-Thought (CoT) prompting, enhancing complex analysis and decision-making capacity. Market Forecasting Agents, Document Analysis Agents, and Trading Strategies Agents utilize CoT to dissect financial challenges into logical steps, aligning their advanced algorithms and domain expertise with the evolving dynamics of financial markets for precise, actionable insights.
  2. Financial LLMs Algorithms Layer: The Financial LLMs Algorithms Layer configures and utilizes specially tuned models tailored to specific domains and global market analysis.
  3. LLMOps and DataOps Layers: The LLMOps layer implements a multi-source integration strategy that selects the most suitable LLMs for specific financial tasks, utilizing a range of state-of-the-art models.
  4. Multi-source LLM Foundation Models Layer: This foundational layer supports the plug-and-play functionality of various general and specialized LLMs.

FinRobot: Agent Workflow

  1. Perception: This module captures and interprets multimodal financial data from market feeds, news, and economic indicators, using sophisticated techniques to structure the data for thorough analysis.

  2. Brain: Acting as the core processing unit, this module perceives data from the Perception module with LLMs and utilizes Financial Chain-of-Thought (CoT) processes to generate structured instructions.

  3. Action: This module executes instructions from the Brain module, applying tools to translate analytical insights into actionable outcomes. Actions include trading, portfolio adjustments, generating reports, or sending alerts, thereby actively influencing the financial environment.

FinRobot: Smart Scheduler

The Smart Scheduler is central to ensuring model diversity and optimizing the integration and selection of the most appropriate LLM for each task.

  • Director Agent: This component orchestrates the task assignment process, ensuring that tasks are allocated to agents based on their performance metrics and suitability for specific tasks.
  • Agent Registration: Manages the registration and tracks the availability of agents within the system, facilitating an efficient task allocation process.
  • Agent Adaptor: Tailor agent functionalities to specific tasks, enhancing their performance and integration within the overall system.
  • Task Manager: Manages and stores different general and fine-tuned LLMs-based agents tailored for various financial tasks, updated periodically to ensure relevance and efficacy.

File Structure

The main folder finrobot has three subfolders agents, data_source, functional.

FinRobot
β”œβ”€β”€ finrobot (main folder)
β”‚   β”œβ”€β”€ agents
β”‚   	β”œβ”€β”€ agent_library.py
β”‚   	└── workflow.py
β”‚   β”œβ”€β”€ data_source
β”‚   	β”œβ”€β”€ finnhub_utils.py
β”‚   	β”œβ”€β”€ finnlp_utils.py
β”‚   	β”œβ”€β”€ fmp_utils.py
β”‚   	β”œβ”€β”€ sec_utils.py
β”‚   	└── yfinance_utils.py
β”‚   β”œβ”€β”€ functional
β”‚   	β”œβ”€β”€ analyzer.py
β”‚   	β”œβ”€β”€ charting.py
β”‚   	β”œβ”€β”€ coding.py
β”‚   	β”œβ”€β”€ quantitative.py
β”‚   	β”œβ”€β”€ reportlab.py
β”‚   	└── text.py
β”‚   β”œβ”€β”€ toolkits.py
β”‚   └── utils.py
β”‚
β”œβ”€β”€ configs
β”œβ”€β”€ experiments
β”œβ”€β”€ tutorials_beginner (hands-on tutorial)
β”‚   β”œβ”€β”€ agent_fingpt_forecaster.ipynb
β”‚   └── agent_annual_report.ipynb 
β”œβ”€β”€ tutorials_advanced (advanced tutorials for potential finrobot developers)
β”‚   β”œβ”€β”€ agent_trade_strategist.ipynb
β”‚   β”œβ”€β”€ agent_fingpt_forecaster.ipynb
β”‚   β”œβ”€β”€ agent_annual_report.ipynb 
β”‚   β”œβ”€β”€ lmm_agent_mplfinance.ipynb
β”‚   └── lmm_agent_opt_smacross.ipynb
β”œβ”€β”€ setup.py
β”œβ”€β”€ OAI_CONFIG_LIST_sample
β”œβ”€β”€ config_api_keys_sample
β”œβ”€β”€ requirements.txt
└── README.md

Installation:

1. (Recommended) Create a new virtual environment

conda create --name finrobot python=3.10
conda activate finrobot

2. download the FinRobot repo use terminal or download it manually

git clone https://github.com/AI4Finance-Foundation/FinRobot.git
cd FinRobot

3. install finrobot & dependencies from source or pypi

get our latest release from pypi

pip install -U finrobot

or install from this repo directly

pip install -e .

4. modify OAI_CONFIG_LIST_sample file

1) rename OAI_CONFIG_LIST_sample to OAI_CONFIG_LIST
2) remove the four lines of comment within the OAI_CONFIG_LIST file
3) add your own openai api-key <your OpenAI API key here>

5. modify config_api_keys_sample file

1) rename config_api_keys_sample to config_api_keys
2) remove the comment within the config_api_keys file
3) add your own finnhub-api "YOUR_FINNHUB_API_KEY"
4) add your own financialmodelingprep and sec-api keys "YOUR_FMP_API_KEY" and "YOUR_SEC_API_KEY" (for financial report generation)

6. start navigating the tutorials or the demos below:

# find these notebooks in tutorials
1) agent_annual_report.ipynb
2) agent_fingpt_forecaster.ipynb
3) agent_trade_strategist.ipynb
4) lmm_agent_mplfinance.ipynb
5) lmm_agent_opt_smacross.ipynb

Demos

1. Market Forecaster Agent (Predict Stock Movements Direction)

Takes a company's ticker symbol, recent basic financials, and market news as input and predicts its stock movements.

  1. Import
import autogen
from finrobot.utils import get_current_date, register_keys_from_json
from finrobot.agents.workflow import SingleAssistant
  1. Config
# Read OpenAI API keys from a JSON file
llm_config = {
    "config_list": autogen.config_list_from_json(
        "../OAI_CONFIG_LIST",
        filter_dict={"model": ["gpt-4-0125-preview"]},
    ),
    "timeout": 120,
    "temperature": 0,
}

# Register FINNHUB API keys
register_keys_from_json("../config_api_keys")
  1. Run
company = "NVDA"

assitant = SingleAssistant(
    "Market_Analyst",
    llm_config,
    # set to "ALWAYS" if you want to chat instead of simply receiving the prediciton
    human_input_mode="NEVER",
)
assitant.chat(
    f"Use all the tools provided to retrieve information available for {company} upon {get_current_date()}. Analyze the positive developments and potential concerns of {company} "
    "with 2-4 most important factors respectively and keep them concise. Most factors should be inferred from company related news. "
    f"Then make a rough prediction (e.g. up/down by 2-3%) of the {company} stock price movement for next week. Provide a summary analysis to support your prediction."
)
  1. Result

2. Financial Analyst Agent for Report Writing (Equity Research Report)

Take a company's 10-k form, financial data, and market data as input and output an equity research report

  1. Import
import os
import autogen
from textwrap import dedent
from finrobot.utils import register_keys_from_json
from finrobot.agents.workflow import SingleAssistantShadow
  1. Config
llm_config = {
    "config_list": autogen.config_list_from_json(
        "../OAI_CONFIG_LIST",
        filter_dict={
            "model": ["gpt-4-0125-preview"],
        },
    ),
    "timeout": 120,
    "temperature": 0.5,
}
register_keys_from_json("../config_api_keys")

# Intermediate strategy modules will be saved in this directory
work_dir = "../report"
os.makedirs(work_dir, exist_ok=True)

assistant = SingleAssistantShadow(
    "Expert_Investor",
    llm_config,
    max_consecutive_auto_reply=None,
    human_input_mode="TERMINATE",
)
  1. Run
company = "Microsoft"
fyear = "2023"

message = dedent(
    f"""
    With the tools you've been provided, write an annual report based on {company}'s {fyear} 10-k report, format it into a pdf.
    Pay attention to the followings:
    - Explicitly explain your working plan before you kick off.
    - Use tools one by one for clarity, especially when asking for instructions. 
    - All your file operations should be done in "{work_dir}". 
    - Display any image in the chat once generated.
    - All the paragraphs should combine between 400 and 450 words, don't generate the pdf until this is explicitly fulfilled.
"""
)

assistant.chat(message, use_cache=True, max_turns=50,
               summary_method="last_msg")
  1. Result

Financial CoT:

  1. Gather Preliminary Data: 10-K report, market data, financial ratios
  2. Analyze Financial Statements: balance sheet, income statement, cash flow
  3. Company Overview and Performance: company description, business highlights, segment analysis
  4. Risk Assessment: assess risks
  5. Financial Performance Visualization: plot PE ratio and EPS
  6. Synthesize Findings into Paragraphs: combine all parts into a coherent summary
  7. Generate PDF Report: use tools to generate PDF automatically
  8. Quality Assurance: check word counts

3. Trade Strategist Agent with multimodal capabilities

AI Agent Papers

AI Agent Blogs and Videos

AI Agent Open-Source Framework & Tool

  • AutoGPT (161k stars) is a tool for everyone to use, aiming to democratize AI, making it accessible for everyone to use and build upon.
  • LangChain (82.7k stars) is a framework for developing context-aware applications powered by language models, enabling them to connect to sources of context and rely on the model's reasoning capabilities for responses and actions.
  • MetaGPT (39.1k stars) is a multi-agent open-source framework that assigns different roles to GPTs, forming a collaborative software entity to execute complex tasks.
  • AutoGen (24.8k stars) is a framework for developing LLM applications with conversational agents that collaborate to solve tasks. These agents are customizable, support human interaction, and operate in modes combining LLMs, human inputs, and tools.
  • dify (22.7k stars) is an LLM application development platform. It integrates the concepts of Backend as a Service and LLMOps, covering the core tech stack required for building generative AI-native applications, including a built-in RAG engine
  • ChatDev (22.7k stars) is a framework that focuses on developing conversational AI Agents capable of dialogue and question-answering. It provides a range of pre-trained models and interactive interfaces, facilitating the development of customized chat Agents for users.
  • BabyAGI (19.2k stars) is an AI-powered task management system, dedicated to building AI Agents with preliminary general intelligence.
  • SuperAGI (14.4k stars) is a dev-first open-source autonomous AI agent framework enabling developers to build, manage & run useful autonomous agents.
  • FastGPT (12.5k stars) is a knowledge-based platform built on the LLM, offers out-of-the-box data processing and model invocation capabilities, allows for workflow orchestration through Flow visualization.
  • CrewAI (12.1k stars) is a framework for orchestrating role-playing, autonomous AI agents. By fostering collaborative intelligence, CrewAI empowers agents to work together seamlessly, tackling complex tasks.
  • XAgent (7.5k stars) is an open-source experimental Large Language Model (LLM) driven autonomous agent that can automatically solve various tasks.
  • Bisheng (5.5k stars) is a leading open-source platform for developing LLM applications.
  • Voyager (5.1k stars) An Open-Ended Embodied Agent with Large Language Models.
  • CAMEL (4.4k stars) is a framework that offers a comprehensive set of tools and algorithms for building multimodal AI Agents, enabling them to handle various data forms such as text, images, and speech.
  • Langfuse (2.9k stars) is a language fusion framework that can integrate the language abilities of multiple AI Agents, enabling them to simultaneously possess multilingual understanding and generation capabilities.

Disclaimer: The codes and documents provided herein are released under the Apache-2.0 license. They should not be construed as financial counsel or recommendations for live trading. It is imperative to exercise caution and consult with qualified financial professionals prior to any trading or investment actions.