Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

https://github.com/the-crypt-keeper/can-ai-code

Self-evaluating interview for AI coders
https://github.com/the-crypt-keeper/can-ai-code

ai ggml humaneval langchain llama-cpp llm transformers

Last synced: 4 months ago
JSON representation

Self-evaluating interview for AI coders

Lists

README 

        



# Can AI Code?



![A cute robot working on a laptop](img/can-ai-code-small.png "A cute robot working on a laptop")



A self-evaluating interview for AI coding models.






## Key Ideas



* Interview questions written by humans, test taken by AI

* Inference scripts for all common API providers and CUDA-enabled quantization runtimes

* Sandbox enviroment (Docker-based) for untrusted Python and NodeJS code validation

* Evaluate effects of prompting techniques and sampling parameters on LLM coding performance

* Evaluate LLM coding performance degradation due to quantization



## News



**3/03** Re-evaluate CodeLlama family.  Leaderboard results are now split by task!



**3/02** Evaluate bigcoder/starcoder2 family.  Starting to split results into `Instruct`, `Code-Completion` and `Fill-in-the-Middle` objectives.  Upgraded to modal==0.58.92



**2/22** Evaluate google/gemma-it family (neither model can pass junior, senior is a write-off).



**2/19** Evaluate ajibawa-2023/Code-290k-13B. Evaluate internlm/internlm2-chat family (1.8B, 7B, 20B).



**2/18** Re-evaluate Speechless-Codallama-v2.0 with better prompt.



**2/15** Evaluate top end of the Qwen1.5 family (4B, 7B, 14B, 72B).



**2/06** Evaluate upstage/SOLAR-10.7B-Instruct-v1.0.



## Test Suites



`junior-v2` is a multi-language (Python, JavaScript) suite of 12 tests created for this project to test small LLM coding performance.  This project provides all necessary components to execute this evaluation.



:construction: `humaneval` is a Python-only suite of 164 tests created by OpenAI.  This project provides template scripts to prepare and execute the humaneval interview, as well as result extraction scripts to help their evaluator. See https://github.com/openai/human-eval for more information.



## [Click to see Leaderboard on HF Spaces](https://huggingface.co/spaces/mike-ravkine/can-ai-code-results)



## [Click to see Comparisons on HF Spaces](https://huggingface.co/spaces/mike-ravkine/can-ai-code-compare)



### Results data



All model answers and evaluation results are now included inside this repository!  Install a recent release of streamlit `pip install streamlit==1.23` then `streamlit run app.py` or `streamlit run compare-app.py` to run the above webapps locally.



### Results HumanEval



:construction: [humaneval/](humaneval/) development work is currently paused, there's other projects that are much further along.



See https://github.com/my-other-github-account/llm-humaneval-benchmarks and https://github.com/abacaj/code-eval for large lists of Humaneval LLM benchmark results.



## Repository Structure



### Interviews



* `junior-v2/*.yaml` - junior coder interview questions (stable)

* `senior/*.yaml` - senior coder interview questions (WIP)



### Prepare



* `prompts/*.txt` - LLM prompt templates for the various models

* `prepare.py` - Applies templates to question turning them into language- and model-specific prompts suitable for interview



#### Prompts



See [prompts/](prompts/) for all prompts references in the leaderboard.



### Interview



* `params/*.json` - Sampling hyper-parameter sets (used by all interview scripts)

* `interview-*.py` - Interview scripts



#### Parameters



See [params/](params/) for all params references in the leaderboard.



### Evaluate



* `evaluate.py` - Run tests for the generated code in a sandbox and grades each answer

* `app.py` - Streamlit webapp to explore results, see https://huggingface.co/spaces/mike-ravkine/can-ai-code-results



### Compare



* `compare.py` - Performs comparisons between evaluations, optionally calling out to an LLM for analysis

* `compare-app.py` - Streamlit webapp to explore comparisons, see https://huggingface.co/spaces/mike-ravkine/can-ai-code-compare

* `compare/*.yaml` - Compare configurations

* `compare/*.json` - Compare results



## Interviewers: API



| API Runtime              | Script         |

|--------------------------|----------------|

| LiteLLM (OpenAI, etc..)  | `interview-litellm.py` |

| OobaBooga/KoboldCpp      | `interview-oobabooga.py` |

| Huggingface Inference    | `interview-hfinference.py` |

| Gradio (HF Spaces)       | `interview-gradio.py` |



## Interviewers: CUDA (Local)



| Quantization Type        | Script                  | Dependency              |

|--------------------------|-------------------------|-------------------------|

| GGUF                     | `interview-llamacpp.py` | llamacpp or ggml binary |

| GPTQ (AutoGptQ)          | `interview-cuda.py`     | auto-gptq==0.6.0        |

| GPTQ (ExLlama)           | `interview-cuda.py`     | exllama @ 3b013cd53c7d413cf99ca04c7c28dd5c95117c0d |

| EXL2, GPTQ (ExLlama2)    | `interview-cuda.py`     | exllamav2 @ 0.0.12 |

| HQQ                      | `interview-cuda.py`     | hqq @ 0.1.1             |

| AWQ, FP16 (vLLM)         | `interview-cuda.py`     | vllm==0.3.0             |

| CTranslate2              | `interview-cuda.py`     | ctranslate2>=3.16.0     |

| bitsandbytes             | `interview-cuda.py`     | bitsandbytes==0.41.3    |

| FP16 (Transformers)      | `interview-cuda.py`     | transformers==4.37.2    |



### Running on Modal



The recommended modal wrapper is `interview_modal_cuda11.py` which builds a CUDA11.8 based container with all the above dependencies working. An `interview_modal_cuda12.py` is also provided, but AutoGPTQ and CTranslate2 are not compatible.



Unfortunately the nature of Modal does not allow command-line selection of eitehr LLM model or runtime engine.



To select models, open the script and uncomment the `.run_function(download...)` line of choice.  Note that only one model can be selected at a time.   To add a new model, implement a new `download...` function.



To select runtime, open the script and uncomment one of the `RUNTIME` options. Note that for `transformers` you must also specify `QUANT`.



## Question Format



A set of interview questions is a folder of .yaml files.  Each Question is a top-level key:



```yaml

SanityList:

    Signature: "things()"

    Input: "with no inputs"

    Output: "a list with three values: the number 5, the string 'foobar', the capital city of Spain"

    Fact: "the capital city of Spain is Madrid"

    Description: "List function, see if the model can combine input facts with internal knowledge."

    Checks:

        input_name:

            assert: "f.name"

            eq: "things"

```



In this example `SanityList` is the name of the interview question.



The first four fields are used by `prepare.py` to create the interview:



- `Signature` is the desired function signature

- `Input` describes the function inputs

- `Output` describes the function outputs

- `Fact` is optional and provides any context that is required to correctly perform the task



These 4 variables along with `language` (either `python` or `javascript`) are used to expand templates in `prompts/`.



The last two fields are used by `evaluate.py` to judge the results:



- `Description` is a human-readable explanation of why this test is useful

- `Checks` defines the expected behavior of the output.



### Checks and the 'f' object



Each check has a name, some `assert` value (python code) and an expected `eq` value.



The f object represents the sandbox view of the function.  Static analysis is performed on the function signature to extract the `f.name` and `f.args` fields, while `f.call` allows for function evaluation.



## Output formats



All scripts output automatically named .ndjson files to the `results/` directory.



Each stage outputs a super-set of fields from the stage before it, so its possible to feed eval/interview back to interview (to re-run the questions) or back to eval (to re-run the eval).



### prepare



`results/prepare_{interview}_{languages}_{template}.ndjson`



Fields:



- all Question fields (Signature, Input, Output, Fact, Description)

- name

- language

- prompt



### interview



`results/interview_{interview}_{languages}_{template}_{templateout}_{params}_{model}_{timestamp}.ndjson`



Fields:

- all `prepare` fields

- model

- params

- answer

- runtime



### eval



`results/eval_{interview}_{languages}_{template}_{templateout}_{params}_{model}_{timestamp}.ndjson`



Fields:

- all `eval` fields

- status

- passed

- total

- checks



# Roadmap / Future Work



* [Development of a Senior coder test suite](https://github.com/the-crypt-keeper/can-ai-code/issues/141)

* Open [Model Request](https://github.com/the-crypt-keeper/can-ai-code/labels/model%20request) issues