├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── llm_query_expansion ├── __init__.py ├── __main__.py ├── evaluate.py ├── expansion.py └── model_wrappers.py ├── misc ├── llm-query-expansion.png ├── prompt-diff.png └── prompt.png ├── pyproject.toml └── query_expansions ├── fiqa-gemini-flash-100-special.json ├── fiqa-gemini-flash-100.json ├── fiqa-gemini-flash-150-special.json ├── fiqa-gemini-flash-150.json ├── fiqa-gemini-flash-250-special.json ├── fiqa-gemini-flash-250.json ├── nfcorpus-gemini-flash-100.json ├── nfcorpus-gemini-flash-150.json ├── nfcorpus-gemini-flash-250.json ├── scifact-gemini-flash-100-special.json ├── scifact-gemini-flash-100.json ├── scifact-gemini-flash-150-special.json ├── scifact-gemini-flash-150.json ├── scifact-gemini-flash-250-special.json ├── scifact-gemini-flash-250.json ├── touche-gemini-flash-100.json ├── touche-gemini-flash-150.json ├── touche-gemini-flash-250.json ├── treccovid-gemini-flash-100.json ├── treccovid-gemini-flash-150.json └── treccovid-gemini-flash-250.json /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/README.md -------------------------------------------------------------------------------- /llm_query_expansion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /llm_query_expansion/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/llm_query_expansion/__main__.py -------------------------------------------------------------------------------- /llm_query_expansion/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/llm_query_expansion/evaluate.py -------------------------------------------------------------------------------- /llm_query_expansion/expansion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/llm_query_expansion/expansion.py -------------------------------------------------------------------------------- /llm_query_expansion/model_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/llm_query_expansion/model_wrappers.py -------------------------------------------------------------------------------- /misc/llm-query-expansion.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/misc/llm-query-expansion.png -------------------------------------------------------------------------------- /misc/prompt-diff.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/misc/prompt-diff.png -------------------------------------------------------------------------------- /misc/prompt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/misc/prompt.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/pyproject.toml -------------------------------------------------------------------------------- /query_expansions/fiqa-gemini-flash-100-special.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/fiqa-gemini-flash-100-special.json -------------------------------------------------------------------------------- /query_expansions/fiqa-gemini-flash-100.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/fiqa-gemini-flash-100.json -------------------------------------------------------------------------------- /query_expansions/fiqa-gemini-flash-150-special.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/fiqa-gemini-flash-150-special.json -------------------------------------------------------------------------------- /query_expansions/fiqa-gemini-flash-150.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/fiqa-gemini-flash-150.json -------------------------------------------------------------------------------- /query_expansions/fiqa-gemini-flash-250-special.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/fiqa-gemini-flash-250-special.json -------------------------------------------------------------------------------- /query_expansions/fiqa-gemini-flash-250.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/fiqa-gemini-flash-250.json -------------------------------------------------------------------------------- /query_expansions/nfcorpus-gemini-flash-100.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/nfcorpus-gemini-flash-100.json -------------------------------------------------------------------------------- /query_expansions/nfcorpus-gemini-flash-150.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/nfcorpus-gemini-flash-150.json -------------------------------------------------------------------------------- /query_expansions/nfcorpus-gemini-flash-250.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/nfcorpus-gemini-flash-250.json -------------------------------------------------------------------------------- /query_expansions/scifact-gemini-flash-100-special.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/scifact-gemini-flash-100-special.json -------------------------------------------------------------------------------- /query_expansions/scifact-gemini-flash-100.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/scifact-gemini-flash-100.json -------------------------------------------------------------------------------- /query_expansions/scifact-gemini-flash-150-special.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/scifact-gemini-flash-150-special.json -------------------------------------------------------------------------------- /query_expansions/scifact-gemini-flash-150.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/scifact-gemini-flash-150.json -------------------------------------------------------------------------------- /query_expansions/scifact-gemini-flash-250-special.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/scifact-gemini-flash-250-special.json -------------------------------------------------------------------------------- /query_expansions/scifact-gemini-flash-250.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/scifact-gemini-flash-250.json -------------------------------------------------------------------------------- /query_expansions/touche-gemini-flash-100.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/touche-gemini-flash-100.json -------------------------------------------------------------------------------- /query_expansions/touche-gemini-flash-150.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/touche-gemini-flash-150.json -------------------------------------------------------------------------------- /query_expansions/touche-gemini-flash-250.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/touche-gemini-flash-250.json -------------------------------------------------------------------------------- /query_expansions/treccovid-gemini-flash-100.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/treccovid-gemini-flash-100.json -------------------------------------------------------------------------------- /query_expansions/treccovid-gemini-flash-150.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/treccovid-gemini-flash-150.json -------------------------------------------------------------------------------- /query_expansions/treccovid-gemini-flash-250.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jina-ai/llm-query-expansion/HEAD/query_expansions/treccovid-gemini-flash-250.json --------------------------------------------------------------------------------