Files
windmill/llm
Ruben Fiszel 660fb549fb fix: llm/requirements.txt to reduce vulnerabilities (#3965)
The following vulnerabilities are fixed by pinning transitive dependencies:
- https://snyk.io/vuln/SNYK-PYTHON-ANYIO-7361842

Co-authored-by: snyk-bot <snyk-bot@snyk.io>
2024-09-01 23:51:10 +02:00
..

Test sample queries of the AI feature

Setup

Create a virtual environment python -m venv .venv, activate it source .venv/bin/activate and install the requirements pip install -r requirements.txt. Put your OPENAI_API_KEY in .env.

Run

Run the script python src/gen_samples.py and check the results in sample_answers.yaml.

Add queries

You can add more queries in sample_queries.yaml. There are three types of queries:

Generate code from a prompt

- type: gen
  description: hello world
  lang: python3

Modifiy code according to a prompt

- type: edit
  description: comment
  lang: python3
  code: |-
    print("hello world")

Fix code given an error

- type: fix
  lang: python3
  code: |-
    def main():
        return 3 / 0
  error: division by zero