From e03d468fa5cbdca96991f00eaf63e44e86818869 Mon Sep 17 00:00:00 2001 From: Nicola Di Cicco <93935338+nicoladicicco@users.noreply.github.com> Date: Tue, 24 Sep 2024 15:13:52 +0900 Subject: [PATCH] Update README.md --- README.md | 28 +++++++++++++++++++--------- 1 file changed, 19 insertions(+), 9 deletions(-) diff --git a/README.md b/README.md index a8fcf06..e0492da 100644 --- a/README.md +++ b/README.md @@ -4,7 +4,7 @@ [![Coverage](https://codecov.io/gh/Azzaare/ConstraintsTranslator.jl/branch/main/graph/badge.svg)](https://codecov.io/gh/Azzaare/ConstraintsTranslator.jl) [![Aqua](https://raw.githubusercontent.com/JuliaTesting/Aqua.jl/master/badge.svg)](https://github.com/JuliaTesting/Aqua.jl) -A package for translating natural-language descriptions of optimization problems into Constraint Programming models to be solved via [`CBLS.jl`](https://github.com/JuliaConstraints/CBLS.jl) using Large Language Models (LLMs). +A package for translating natural-language descriptions of optimization problems into Constraint Programming models using Large Language Models (LLMs). For this pre-stable version stage, our target is to have models solved via [`CBLS.jl`](https://github.com/JuliaConstraints/CBLS.jl). Eventually, we expect this library to work for most of Julia CP ecosystem, alongside other CP modeling languages such as MiniZinc, OR-Tools, etc. This package acts as a light wrapper around common LLM API endpoints, supplying appropriate system prompts and context informations to the LLMs to generate CP models. Specifically, we first prompt the model for generating an high-level representation of the problem in editable Markdown format, and then we prompt the model to generate Julia code. @@ -17,17 +17,28 @@ We currently support the following LLM APIs: Groq and Gemini are currently offering rate-limited free access to their APIs, and llama.cpp is free and open-source. We are still actively experimenting with this package, and we are not in a position to pay for API access. We might consider adding support for other APIs in the future. ## Workflow example -To begin playing with the package, you can start from the example below: +Before playing with the package, we need to set up two environment variables: +1. The EDITOR variable for specifying a text editor (such as `vim`, `nano`, `emacs`, ...). This will be used during interactive execution. +2. An API key. This is necessary only for interacting with proprietary LLMs. + +We can configure those variables by, e.g., appending the following to your `.bashrc` or equivalent: +```bash +export EDITOR="vim" +export GOOGLE_API_KEY="42" +``` + +Or we can configure them in Julia: +```julia +ENV["EDITOR"] = "vim" +ENV["GOOGLE_API_KEY"] = "42" +``` +Finally, we can start playing with the package. Below, an example for translating a natural-language description of the Traveling Salesman Problem: ```julia using ConstraintsTranslator llm = GoogleLLM("gemini-1.5-pro") -# Optional setup of a terminal editor (uncomment and select a viable editor on your machine such as vim, nano, emacs, ...) -ENV["EDITOR"] = "vim" - - description = """ We need to determine the shortest possible route for a salesman who must visit a set of cities exactly once and return to the starting city. The objective is to minimize the total travel distance while ensuring that each city is visited exactly once. @@ -44,16 +55,15 @@ CityA,CityB,10 CityA,CityC,8 """ -response = translate(llm, description, interactive=true) +response = translate(llm, description) ``` The `translate` function will first produce a Markdown representation of the problem, and then return the generated Julia code for parsing the input data and building the model. The flag `interactive=true` will enable a simple interactive command-line application, where you will be able to inspect, edit and regenerate each intermediate output. -This example uses Google Gemini as an LLM. You will need an API key and a model id to access proprietary API endpoints. Use `help?>` in the Julia REPL to learn more about the available models. - At each generation step, it will prompt the user in an interactive menu to accept the answer, edit the prompt and/or the generated text, or generate another answer with the same prompt. The LLM expects the user to provide examples of the input data format. If no examples are present, the LLM will make assumptions about the data format based on the problem description. +This example uses Google Gemini as an LLM. You will need an API key and a model id to access proprietary API endpoints. Use `help?>` in the Julia REPL to learn more about the available models. \ No newline at end of file