cyyeh / data-agent Public

Notifications You must be signed in to change notification settings
Fork 1
Star 0

This repository contains the data agent code for the DABstep benchmark.

0 stars 1 fork Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 27 Commits
.env.example		.env.example
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
baseline.ipynb		baseline.ipynb
main.py		main.py
poetry.lock		poetry.lock
pyproject.toml		pyproject.toml
utils.py		utils.py

Repository files navigation

Data Agent

This repository contains the data agent code for the DABstep benchmark.

Preparation

install poetry
prepare the environment: poetry install
prepare the environment variables: copying .env.example to .env and filling in environment variables
- HF_TOKEN: make sure to create a token with read access to the DABstep dataset, DABstep space and inference API
- OPENAI_API_KEY: OpenAI API key if you want to use OpenAI model
- LANGFUSE_PUBLIC_KEY: Langfuse public key
- LANGFUSE_SECRET_KEY: Langfuse secret key
- LANGFUSE_HOST: Langfuse host

Run the agent

poetry run python main.py

References