Skip to content

Improved prompting and task type decisions for multi agent model #39

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 4 commits into
base: ltsm-stack
Choose a base branch
from

Conversation

joshhan619
Copy link
Contributor

Planning_Agent.py:

  • Improved the LLM prompts to get better answers from LLM
  • Planning agent decides on best task type for TS Agent based on task description

TS_agent.py:

  • Different behavior depending on task type
  • If task type is classification, use a LLM to describe time series data
  • If task type is forecasting, use time-series model (e.g. LTSM-bundle) to predict and then send description of predicted time-series to QA

QA_Agent.py:

  • Changed prompt for LLM depending on task type

ltsm_inference.py:

  • Changed the input of LTSM to use pretrained weights leading to improved forecasting results
  • Added padding for the prompt data

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant