├── .coveragerc ├── .github ├── actions │ ├── free-disk-space │ │ └── action.yml │ └── ollama-setup │ │ └── action.yml ├── dependabot.yml └── workflows │ ├── actionlint.dockerfile │ ├── actionlint.yml │ ├── lint.yml │ ├── matchers │ ├── actionlint.json │ └── pylint.json │ ├── pypi.yml │ ├── test.yml │ ├── test_contrib.yml │ ├── test_e2e.yml │ ├── test_e2e_3_3.yml │ └── test_examples.yml ├── .gitignore ├── .isort.cfg ├── .pre-commit-config.yaml ├── .pylintrc ├── CONTRIBUTING.md ├── LICENSE ├── OWNERS ├── README.md ├── code-of-conduct.md ├── contrib ├── 00_example │ ├── .pylintrc │ ├── README.md │ ├── examples │ │ └── hello.py │ ├── notebooks │ │ └── hello.ipynb │ ├── requirements.txt │ ├── src │ │ ├── __init__.py │ │ └── hello_lib.py │ ├── tests │ │ └── test_hello.py │ └── tox.ini └── pdl │ ├── .pylintrc │ ├── README.md │ ├── examples │ └── enforce_low_hallucination.py │ ├── notebooks │ ├── pdl.ipynb │ └── sequential_scaling.ipynb │ ├── pyproject.toml │ ├── src │ └── granite_io_pdl │ │ ├── pdl_io.py │ │ ├── sequential_scaling.pdl │ │ └── sequential_scaling.py │ ├── tests │ └── io │ │ └── test_pdl_io.py │ └── tox.ini ├── data ├── .gitignore └── test_retrieval │ ├── govt10.jsonl.zip │ └── govt10_embeds.parquet │ └── part_001.parquet ├── examples ├── model_chat.py ├── model_chat_with_citation.py ├── model_chat_with_citation_hallucination.py ├── model_chat_with_mbrd_majority_voting.py ├── model_chat_with_thinking.py ├── model_chat_with_thinking_custom_io_processor.py ├── model_chat_with_thinking_separate_backend.py ├── rerank_with_llm.py └── watsonx_litellm.py ├── mkdocs ├── README.md ├── docs │ ├── api.md │ ├── design.md │ ├── img │ │ ├── favicon.ico │ │ ├── granite-io-full-architecture.png │ │ ├── granite-io-input-output-architecture.png │ │ ├── granite-io-io-proc-architecture.png │ │ └── ibm_logo_rebus.png │ └── index.md └── mkdocs.yml ├── notebooks ├── .gitignore ├── answer_relevance.ipynb ├── answerability.ipynb ├── certainty.ipynb ├── citations.ipynb ├── context_relevancy.ipynb ├── hallucinations.ipynb ├── hyde_rewrite.ipynb ├── intrinsics.ipynb ├── io.ipynb ├── mbrd.ipynb ├── pdl_demo.ipynb ├── prm_bestofn.ipynb ├── query_expansion.ipynb ├── query_rewrite.ipynb ├── rag.ipynb ├── retrieval_elasticsearch.ipynb ├── retrieval_embeddings.ipynb └── voting.ipynb ├── pyproject.toml ├── pytest.ini ├── src └── granite_io │ ├── __init__.py │ ├── backend │ ├── __init__.py │ ├── base.py │ ├── litellm.py │ ├── openai.py │ ├── registry.py │ ├── transformers.py │ └── vllm_server.py │ ├── factory.py │ ├── io │ ├── __init__.py │ ├── answer_relevance │ │ ├── __init__.py │ │ └── answer_relevance.py │ ├── answerability │ │ ├── __init__.py │ │ └── answerability.py │ ├── base.py │ ├── certainty │ │ ├── __init__.py │ │ └── certainty.py │ ├── citations │ │ ├── __init__.py │ │ └── citations.py │ ├── consts.py │ ├── context_relevancy │ │ ├── __init__.py │ │ └── context_relevancy.py │ ├── granite_3_2 │ │ ├── __init__.py │ │ ├── granite_3_2.py │ │ ├── input_processors │ │ │ ├── __init__.py │ │ │ └── granite_3_2_input_processor.py │ │ └── output_processors │ │ │ ├── __init__.py │ │ │ ├── granite_3_2_output_parser.py │ │ │ └── granite_3_2_output_processor.py │ ├── granite_3_3 │ │ ├── __init__.py │ │ ├── granite_3_3.py │ │ ├── input_processors │ │ │ ├── __init__.py │ │ │ └── granite_3_3_input_processor.py │ │ └── output_processors │ │ │ ├── __init__.py │ │ │ ├── granite_3_3_output_parser.py │ │ │ └── granite_3_3_output_processor.py │ ├── hallucinations │ │ ├── __init__.py │ │ └── hallucinations.py │ ├── hyde_rewrite │ │ ├── __init__.py │ │ └── hyde_rewrite.py │ ├── llmrerank │ │ ├── README.md │ │ ├── __init__.py │ │ ├── images │ │ │ └── Picture1.png │ │ └── llmrerank.py │ ├── process_reward_model │ │ ├── __init__.py │ │ └── best_of_n.py │ ├── query_expansion │ │ ├── __init__.py │ │ └── query_expansion.py │ ├── query_rewrite │ │ ├── __init__.py │ │ └── query_rewrite.py │ ├── rag_agent_lib │ │ ├── __init__.py │ │ └── util.py │ ├── registry.py │ ├── retrieval │ │ ├── __init__.py │ │ ├── retrieval.py │ │ └── util.py │ └── voting │ │ ├── __init__.py │ │ ├── mbrd.py │ │ └── simple.py │ ├── optional.py │ ├── types.py │ └── visualization │ ├── __init__.py │ ├── citations.py │ ├── index.css │ └── index.js ├── tests ├── cassettes │ └── test_backends │ │ ├── test_num_return_sequences_1[backend_litellm].yaml │ │ ├── test_num_return_sequences_1[backend_openai].yaml │ │ ├── test_num_return_sequences_3[backend_litellm].yaml │ │ ├── test_num_return_sequences_3[backend_openai].yaml │ │ ├── test_num_return_sequences_invalid[backend_litellm--1].yaml │ │ ├── test_num_return_sequences_invalid[backend_litellm-0].yaml │ │ ├── test_simple[backend_litellm].yaml │ │ └── test_simple[backend_openai].yaml ├── conftest.py ├── io │ ├── cassettes │ │ ├── test_answer_relevance_classifier │ │ │ ├── test_run_classifier[example1].yaml │ │ │ ├── test_run_classifier[example2].yaml │ │ │ └── test_run_classifier[example3].yaml │ │ ├── test_answer_relevance_composite │ │ │ ├── test_run_composite[example1].yaml │ │ │ ├── test_run_composite[example2].yaml │ │ │ └── test_run_composite[example3].yaml │ │ ├── test_answer_relevance_rewriter │ │ │ ├── test_run_rewriter[example1].yaml │ │ │ └── test_run_rewriter[example2].yaml │ │ ├── test_answerability │ │ │ └── test_run_model.yaml │ │ ├── test_certainty │ │ │ ├── test_run_composite.yaml │ │ │ └── test_run_model.yaml │ │ ├── test_citations │ │ │ ├── test_run_composite.yaml │ │ │ └── test_run_model.yaml │ │ ├── test_context_relevancy │ │ │ └── test_run_model.yaml │ │ ├── test_granite_3_2 │ │ │ ├── test_run_processor[backend_litellm-simple].yaml │ │ │ ├── test_run_processor[backend_litellm-thinking_tag].yaml │ │ │ ├── test_run_processor[backend_openai-simple].yaml │ │ │ └── test_run_processor[backend_openai-thinking_tag].yaml │ │ ├── test_hallucinations │ │ │ ├── test_run_composite.yaml │ │ │ ├── test_run_model.yaml │ │ │ └── test_run_processor.yaml │ │ ├── test_make_io_processor │ │ │ ├── test_make_io_processor[backend_litellm-simple].yaml │ │ │ ├── test_make_io_processor[backend_litellm-stop_strings].yaml │ │ │ ├── test_make_io_processor[backend_litellm-thinking_tag].yaml │ │ │ ├── test_make_io_processor[backend_openai-simple].yaml │ │ │ ├── test_make_io_processor[backend_openai-stop_strings].yaml │ │ │ └── test_make_io_processor[backend_openai-thinking_tag].yaml │ │ ├── test_prm │ │ │ ├── test_run_composite.yaml │ │ │ └── test_run_model.yaml │ │ ├── test_query_expansion │ │ │ └── test_run_model.yaml │ │ ├── test_query_rewrite │ │ │ ├── test_request_processor.yaml │ │ │ └── test_run_model.yaml │ │ ├── test_rerank │ │ │ └── test_rerank_request_processor.yaml │ │ └── test_voting │ │ │ ├── test_mbrd_majority_voting[backend_openai].yaml │ │ │ └── test_numeric_voting[backend_openai].yaml │ ├── data │ │ └── test_answer_relevance3.yaml │ ├── granite_3_2 │ │ ├── cassettes │ │ │ └── test_granite_3_2 │ │ │ │ ├── test_completion_presence_param[backend_litellm].yaml │ │ │ │ ├── test_completion_presence_param[backend_openai].yaml │ │ │ │ ├── test_completion_repetition_param[backend_litellm].yaml │ │ │ │ ├── test_completion_repetition_param[backend_openai].yaml │ │ │ │ ├── test_multiple_return[backend_openai-custom_system_prompt].yaml │ │ │ │ ├── test_multiple_return[backend_openai-functions].yaml │ │ │ │ ├── test_multiple_return[backend_openai-simple].yaml │ │ │ │ ├── test_multiple_return[backend_openai-stop_strings].yaml │ │ │ │ ├── test_multiple_return[backend_openai-thinking_tag].yaml │ │ │ │ ├── test_run_processor[backend_litellm-custom_system_prompt].yaml │ │ │ │ ├── test_run_processor[backend_litellm-functions].yaml │ │ │ │ ├── test_run_processor[backend_litellm-simple].yaml │ │ │ │ ├── test_run_processor[backend_litellm-stop_strings].yaml │ │ │ │ ├── test_run_processor[backend_litellm-thinking_tag].yaml │ │ │ │ ├── test_run_processor[backend_openai-custom_system_prompt].yaml │ │ │ │ ├── test_run_processor[backend_openai-functions].yaml │ │ │ │ ├── test_run_processor[backend_openai-simple].yaml │ │ │ │ ├── test_run_processor[backend_openai-stop_strings].yaml │ │ │ │ └── test_run_processor[backend_openai-thinking_tag].yaml │ │ ├── input_processors │ │ │ ├── test_granite_3_2_input_processor.py │ │ │ └── testdata │ │ │ │ ├── test_reasoning_prompt.txt │ │ │ │ ├── test_remove_special_tokens_expected_prompt.txt │ │ │ │ └── test_remove_special_tokens_input_json.txt │ │ ├── output_processors │ │ │ ├── test_granite_3_2_output_parser.py │ │ │ ├── test_granite_3_2_output_processor.py │ │ │ └── testdata │ │ │ │ ├── test_document_source.txt │ │ │ │ ├── test_output.txt │ │ │ │ ├── test_output_with_citation.txt │ │ │ │ ├── test_output_with_citation_from_source.txt │ │ │ │ ├── test_output_with_citation_hallucinations.txt │ │ │ │ ├── test_output_with_colons_citation_text.txt │ │ │ │ ├── test_output_with_invalid_citation.txt │ │ │ │ ├── test_output_with_multiple_citations_per_document.txt │ │ │ │ ├── test_raw_reasoning_output.txt │ │ │ │ ├── test_reasoning_output_processor_response.txt │ │ │ │ └── test_reasoning_output_processor_thought.txt │ │ └── test_granite_3_2.py │ ├── granite_3_3 │ │ ├── cassettes │ │ │ └── test_granite_3_3 │ │ │ │ ├── test_completion_presence_param[backend_3_3_openai].yaml │ │ │ │ ├── test_completion_repetition_param[backend_3_3_litellm].yaml │ │ │ │ ├── test_multiple_return[backend_3_3_openai-custom_system_prompt].yaml │ │ │ │ ├── test_multiple_return[backend_3_3_openai-documents].yaml │ │ │ │ ├── test_multiple_return[backend_3_3_openai-functions].yaml │ │ │ │ ├── test_multiple_return[backend_3_3_openai-simple].yaml │ │ │ │ ├── test_multiple_return[backend_3_3_openai-stop_strings].yaml │ │ │ │ ├── test_multiple_return[backend_3_3_openai-thinking_tag].yaml │ │ │ │ ├── test_run_processor[backend_3_3_litellm-custom_system_prompt].yaml │ │ │ │ ├── test_run_processor[backend_3_3_litellm-documents].yaml │ │ │ │ ├── test_run_processor[backend_3_3_litellm-functions].yaml │ │ │ │ ├── test_run_processor[backend_3_3_litellm-simple].yaml │ │ │ │ ├── test_run_processor[backend_3_3_litellm-stop_strings].yaml │ │ │ │ ├── test_run_processor[backend_3_3_litellm-thinking_tag].yaml │ │ │ │ ├── test_run_processor[backend_3_3_openai-custom_system_prompt].yaml │ │ │ │ ├── test_run_processor[backend_3_3_openai-documents].yaml │ │ │ │ ├── test_run_processor[backend_3_3_openai-functions].yaml │ │ │ │ ├── test_run_processor[backend_3_3_openai-simple].yaml │ │ │ │ ├── test_run_processor[backend_3_3_openai-stop_strings].yaml │ │ │ │ └── test_run_processor[backend_3_3_openai-thinking_tag].yaml │ │ ├── input_processors │ │ │ ├── test_granite_3_3_input_processor.py │ │ │ └── testdata │ │ │ │ ├── test_reasoning_prompt.txt │ │ │ │ ├── test_remove_special_tokens_expected_prompt.txt │ │ │ │ └── test_remove_special_tokens_input_json.txt │ │ ├── output_processors │ │ │ ├── test_granite_3_3_output_parser.py │ │ │ ├── test_granite_3_3_output_processor.py │ │ │ └── testdata │ │ │ │ ├── test_document_source.txt │ │ │ │ ├── test_output.txt │ │ │ │ ├── test_output_with_citation.txt │ │ │ │ ├── test_output_with_citation_from_source.txt │ │ │ │ ├── test_output_with_citation_hallucinations.txt │ │ │ │ ├── test_output_with_colons_citation_text.txt │ │ │ │ ├── test_output_with_invalid_citation.txt │ │ │ │ ├── test_output_with_multiple_citations_per_document.txt │ │ │ │ ├── test_raw_reasoning_output.txt │ │ │ │ ├── test_reasoning_output_processor_response.txt │ │ │ │ └── test_reasoning_output_processor_thought.txt │ │ └── test_granite_3_3.py │ ├── test_answer_relevance_classifier.py │ ├── test_answer_relevance_composite.py │ ├── test_answer_relevance_rewriter.py │ ├── test_answerability.py │ ├── test_certainty.py │ ├── test_citations.py │ ├── test_context_relevancy.py │ ├── test_hallucinations.py │ ├── test_hyde_rewrite.py │ ├── test_make_io_processor.py │ ├── test_prm.py │ ├── test_query_expansion.py │ ├── test_query_rewrite.py │ ├── test_rerank.py │ ├── test_retrieval.py │ └── test_voting.py ├── test_backends.py ├── test_factory.py ├── test_types.py └── test_utils.py └── tox.ini /.coveragerc: -------------------------------------------------------------------------------- 1 | [run] 2 | omit = 3 | src/granite_io/_version.py 4 | 5 | -------------------------------------------------------------------------------- /.github/actions/free-disk-space/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/actions/free-disk-space/action.yml -------------------------------------------------------------------------------- /.github/actions/ollama-setup/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/actions/ollama-setup/action.yml -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/actionlint.dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/workflows/actionlint.dockerfile -------------------------------------------------------------------------------- /.github/workflows/actionlint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/workflows/actionlint.yml -------------------------------------------------------------------------------- /.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/workflows/lint.yml -------------------------------------------------------------------------------- /.github/workflows/matchers/actionlint.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/workflows/matchers/actionlint.json -------------------------------------------------------------------------------- /.github/workflows/matchers/pylint.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/workflows/matchers/pylint.json -------------------------------------------------------------------------------- /.github/workflows/pypi.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/workflows/pypi.yml -------------------------------------------------------------------------------- /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.github/workflows/test_contrib.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/workflows/test_contrib.yml -------------------------------------------------------------------------------- /.github/workflows/test_e2e.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/workflows/test_e2e.yml -------------------------------------------------------------------------------- /.github/workflows/test_e2e_3_3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/workflows/test_e2e_3_3.yml -------------------------------------------------------------------------------- /.github/workflows/test_examples.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.github/workflows/test_examples.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.gitignore -------------------------------------------------------------------------------- /.isort.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.isort.cfg -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/.pylintrc -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/LICENSE -------------------------------------------------------------------------------- /OWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/OWNERS -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/README.md -------------------------------------------------------------------------------- /code-of-conduct.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/code-of-conduct.md -------------------------------------------------------------------------------- /contrib/00_example/.pylintrc: -------------------------------------------------------------------------------- 1 | ../../.pylintrc -------------------------------------------------------------------------------- /contrib/00_example/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /contrib/00_example/examples/hello.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/00_example/examples/hello.py -------------------------------------------------------------------------------- /contrib/00_example/notebooks/hello.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/00_example/notebooks/hello.ipynb -------------------------------------------------------------------------------- /contrib/00_example/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/00_example/requirements.txt -------------------------------------------------------------------------------- /contrib/00_example/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /contrib/00_example/src/hello_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/00_example/src/hello_lib.py -------------------------------------------------------------------------------- /contrib/00_example/tests/test_hello.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/00_example/tests/test_hello.py -------------------------------------------------------------------------------- /contrib/00_example/tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/00_example/tox.ini -------------------------------------------------------------------------------- /contrib/pdl/.pylintrc: -------------------------------------------------------------------------------- 1 | ../../.pylintrc -------------------------------------------------------------------------------- /contrib/pdl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/pdl/README.md -------------------------------------------------------------------------------- /contrib/pdl/examples/enforce_low_hallucination.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/pdl/examples/enforce_low_hallucination.py -------------------------------------------------------------------------------- /contrib/pdl/notebooks/pdl.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/pdl/notebooks/pdl.ipynb -------------------------------------------------------------------------------- /contrib/pdl/notebooks/sequential_scaling.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/pdl/notebooks/sequential_scaling.ipynb -------------------------------------------------------------------------------- /contrib/pdl/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/pdl/pyproject.toml -------------------------------------------------------------------------------- /contrib/pdl/src/granite_io_pdl/pdl_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/pdl/src/granite_io_pdl/pdl_io.py -------------------------------------------------------------------------------- /contrib/pdl/src/granite_io_pdl/sequential_scaling.pdl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/pdl/src/granite_io_pdl/sequential_scaling.pdl -------------------------------------------------------------------------------- /contrib/pdl/src/granite_io_pdl/sequential_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/pdl/src/granite_io_pdl/sequential_scaling.py -------------------------------------------------------------------------------- /contrib/pdl/tests/io/test_pdl_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/pdl/tests/io/test_pdl_io.py -------------------------------------------------------------------------------- /contrib/pdl/tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/contrib/pdl/tox.ini -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/data/.gitignore -------------------------------------------------------------------------------- /data/test_retrieval/govt10.jsonl.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/data/test_retrieval/govt10.jsonl.zip -------------------------------------------------------------------------------- /data/test_retrieval/govt10_embeds.parquet/part_001.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/data/test_retrieval/govt10_embeds.parquet/part_001.parquet -------------------------------------------------------------------------------- /examples/model_chat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/examples/model_chat.py -------------------------------------------------------------------------------- /examples/model_chat_with_citation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/examples/model_chat_with_citation.py -------------------------------------------------------------------------------- /examples/model_chat_with_citation_hallucination.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/examples/model_chat_with_citation_hallucination.py -------------------------------------------------------------------------------- /examples/model_chat_with_mbrd_majority_voting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/examples/model_chat_with_mbrd_majority_voting.py -------------------------------------------------------------------------------- /examples/model_chat_with_thinking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/examples/model_chat_with_thinking.py -------------------------------------------------------------------------------- /examples/model_chat_with_thinking_custom_io_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/examples/model_chat_with_thinking_custom_io_processor.py -------------------------------------------------------------------------------- /examples/model_chat_with_thinking_separate_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/examples/model_chat_with_thinking_separate_backend.py -------------------------------------------------------------------------------- /examples/rerank_with_llm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/examples/rerank_with_llm.py -------------------------------------------------------------------------------- /examples/watsonx_litellm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/examples/watsonx_litellm.py -------------------------------------------------------------------------------- /mkdocs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/mkdocs/README.md -------------------------------------------------------------------------------- /mkdocs/docs/api.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/mkdocs/docs/api.md -------------------------------------------------------------------------------- /mkdocs/docs/design.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/mkdocs/docs/design.md -------------------------------------------------------------------------------- /mkdocs/docs/img/favicon.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/mkdocs/docs/img/favicon.ico -------------------------------------------------------------------------------- /mkdocs/docs/img/granite-io-full-architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/mkdocs/docs/img/granite-io-full-architecture.png -------------------------------------------------------------------------------- /mkdocs/docs/img/granite-io-input-output-architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/mkdocs/docs/img/granite-io-input-output-architecture.png -------------------------------------------------------------------------------- /mkdocs/docs/img/granite-io-io-proc-architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/mkdocs/docs/img/granite-io-io-proc-architecture.png -------------------------------------------------------------------------------- /mkdocs/docs/img/ibm_logo_rebus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/mkdocs/docs/img/ibm_logo_rebus.png -------------------------------------------------------------------------------- /mkdocs/docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/mkdocs/docs/index.md -------------------------------------------------------------------------------- /mkdocs/mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/mkdocs/mkdocs.yml -------------------------------------------------------------------------------- /notebooks/.gitignore: -------------------------------------------------------------------------------- 1 | debug_*.ipynb 2 | -------------------------------------------------------------------------------- /notebooks/answer_relevance.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/answer_relevance.ipynb -------------------------------------------------------------------------------- /notebooks/answerability.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/answerability.ipynb -------------------------------------------------------------------------------- /notebooks/certainty.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/certainty.ipynb -------------------------------------------------------------------------------- /notebooks/citations.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/citations.ipynb -------------------------------------------------------------------------------- /notebooks/context_relevancy.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/context_relevancy.ipynb -------------------------------------------------------------------------------- /notebooks/hallucinations.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/hallucinations.ipynb -------------------------------------------------------------------------------- /notebooks/hyde_rewrite.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/hyde_rewrite.ipynb -------------------------------------------------------------------------------- /notebooks/intrinsics.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/intrinsics.ipynb -------------------------------------------------------------------------------- /notebooks/io.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/io.ipynb -------------------------------------------------------------------------------- /notebooks/mbrd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/mbrd.ipynb -------------------------------------------------------------------------------- /notebooks/pdl_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/pdl_demo.ipynb -------------------------------------------------------------------------------- /notebooks/prm_bestofn.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/prm_bestofn.ipynb -------------------------------------------------------------------------------- /notebooks/query_expansion.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/query_expansion.ipynb -------------------------------------------------------------------------------- /notebooks/query_rewrite.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/query_rewrite.ipynb -------------------------------------------------------------------------------- /notebooks/rag.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/rag.ipynb -------------------------------------------------------------------------------- /notebooks/retrieval_elasticsearch.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/retrieval_elasticsearch.ipynb -------------------------------------------------------------------------------- /notebooks/retrieval_embeddings.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/retrieval_embeddings.ipynb -------------------------------------------------------------------------------- /notebooks/voting.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/notebooks/voting.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/pytest.ini -------------------------------------------------------------------------------- /src/granite_io/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/__init__.py -------------------------------------------------------------------------------- /src/granite_io/backend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/backend/__init__.py -------------------------------------------------------------------------------- /src/granite_io/backend/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/backend/base.py -------------------------------------------------------------------------------- /src/granite_io/backend/litellm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/backend/litellm.py -------------------------------------------------------------------------------- /src/granite_io/backend/openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/backend/openai.py -------------------------------------------------------------------------------- /src/granite_io/backend/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/backend/registry.py -------------------------------------------------------------------------------- /src/granite_io/backend/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/backend/transformers.py -------------------------------------------------------------------------------- /src/granite_io/backend/vllm_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/backend/vllm_server.py -------------------------------------------------------------------------------- /src/granite_io/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/factory.py -------------------------------------------------------------------------------- /src/granite_io/io/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/answer_relevance/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/answer_relevance/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/answer_relevance/answer_relevance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/answer_relevance/answer_relevance.py -------------------------------------------------------------------------------- /src/granite_io/io/answerability/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/answerability/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/answerability/answerability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/answerability/answerability.py -------------------------------------------------------------------------------- /src/granite_io/io/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/base.py -------------------------------------------------------------------------------- /src/granite_io/io/certainty/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/certainty/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/certainty/certainty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/certainty/certainty.py -------------------------------------------------------------------------------- /src/granite_io/io/citations/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/citations/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/citations/citations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/citations/citations.py -------------------------------------------------------------------------------- /src/granite_io/io/consts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/consts.py -------------------------------------------------------------------------------- /src/granite_io/io/context_relevancy/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/context_relevancy/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/context_relevancy/context_relevancy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/context_relevancy/context_relevancy.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_2/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_2/granite_3_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_2/granite_3_2.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_2/input_processors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_2/input_processors/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_2/input_processors/granite_3_2_input_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_2/input_processors/granite_3_2_input_processor.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_2/output_processors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_2/output_processors/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_2/output_processors/granite_3_2_output_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_2/output_processors/granite_3_2_output_parser.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_2/output_processors/granite_3_2_output_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_2/output_processors/granite_3_2_output_processor.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_3/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_3/granite_3_3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_3/granite_3_3.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_3/input_processors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_3/input_processors/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_3/input_processors/granite_3_3_input_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_3/input_processors/granite_3_3_input_processor.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_3/output_processors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_3/output_processors/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_3/output_processors/granite_3_3_output_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_3/output_processors/granite_3_3_output_parser.py -------------------------------------------------------------------------------- /src/granite_io/io/granite_3_3/output_processors/granite_3_3_output_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/granite_3_3/output_processors/granite_3_3_output_processor.py -------------------------------------------------------------------------------- /src/granite_io/io/hallucinations/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/hallucinations/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/hallucinations/hallucinations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/hallucinations/hallucinations.py -------------------------------------------------------------------------------- /src/granite_io/io/hyde_rewrite/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/hyde_rewrite/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/hyde_rewrite/hyde_rewrite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/hyde_rewrite/hyde_rewrite.py -------------------------------------------------------------------------------- /src/granite_io/io/llmrerank/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/llmrerank/README.md -------------------------------------------------------------------------------- /src/granite_io/io/llmrerank/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/llmrerank/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/llmrerank/images/Picture1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/llmrerank/images/Picture1.png -------------------------------------------------------------------------------- /src/granite_io/io/llmrerank/llmrerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/llmrerank/llmrerank.py -------------------------------------------------------------------------------- /src/granite_io/io/process_reward_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/process_reward_model/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/process_reward_model/best_of_n.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/process_reward_model/best_of_n.py -------------------------------------------------------------------------------- /src/granite_io/io/query_expansion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/query_expansion/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/query_expansion/query_expansion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/query_expansion/query_expansion.py -------------------------------------------------------------------------------- /src/granite_io/io/query_rewrite/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/query_rewrite/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/query_rewrite/query_rewrite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/query_rewrite/query_rewrite.py -------------------------------------------------------------------------------- /src/granite_io/io/rag_agent_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/rag_agent_lib/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/rag_agent_lib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/rag_agent_lib/util.py -------------------------------------------------------------------------------- /src/granite_io/io/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/registry.py -------------------------------------------------------------------------------- /src/granite_io/io/retrieval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/retrieval/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/retrieval/retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/retrieval/retrieval.py -------------------------------------------------------------------------------- /src/granite_io/io/retrieval/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/retrieval/util.py -------------------------------------------------------------------------------- /src/granite_io/io/voting/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/voting/__init__.py -------------------------------------------------------------------------------- /src/granite_io/io/voting/mbrd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/voting/mbrd.py -------------------------------------------------------------------------------- /src/granite_io/io/voting/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/io/voting/simple.py -------------------------------------------------------------------------------- /src/granite_io/optional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/optional.py -------------------------------------------------------------------------------- /src/granite_io/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/types.py -------------------------------------------------------------------------------- /src/granite_io/visualization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/visualization/__init__.py -------------------------------------------------------------------------------- /src/granite_io/visualization/citations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/visualization/citations.py -------------------------------------------------------------------------------- /src/granite_io/visualization/index.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/visualization/index.css -------------------------------------------------------------------------------- /src/granite_io/visualization/index.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/src/granite_io/visualization/index.js -------------------------------------------------------------------------------- /tests/cassettes/test_backends/test_num_return_sequences_1[backend_litellm].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/cassettes/test_backends/test_num_return_sequences_1[backend_litellm].yaml -------------------------------------------------------------------------------- /tests/cassettes/test_backends/test_num_return_sequences_1[backend_openai].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/cassettes/test_backends/test_num_return_sequences_1[backend_openai].yaml -------------------------------------------------------------------------------- /tests/cassettes/test_backends/test_num_return_sequences_3[backend_litellm].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/cassettes/test_backends/test_num_return_sequences_3[backend_litellm].yaml -------------------------------------------------------------------------------- /tests/cassettes/test_backends/test_num_return_sequences_3[backend_openai].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/cassettes/test_backends/test_num_return_sequences_3[backend_openai].yaml -------------------------------------------------------------------------------- /tests/cassettes/test_backends/test_num_return_sequences_invalid[backend_litellm--1].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/cassettes/test_backends/test_num_return_sequences_invalid[backend_litellm--1].yaml -------------------------------------------------------------------------------- /tests/cassettes/test_backends/test_num_return_sequences_invalid[backend_litellm-0].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/cassettes/test_backends/test_num_return_sequences_invalid[backend_litellm-0].yaml -------------------------------------------------------------------------------- /tests/cassettes/test_backends/test_simple[backend_litellm].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/cassettes/test_backends/test_simple[backend_litellm].yaml -------------------------------------------------------------------------------- /tests/cassettes/test_backends/test_simple[backend_openai].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/cassettes/test_backends/test_simple[backend_openai].yaml -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/io/cassettes/test_answer_relevance_classifier/test_run_classifier[example1].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_answer_relevance_classifier/test_run_classifier[example1].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_answer_relevance_classifier/test_run_classifier[example2].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_answer_relevance_classifier/test_run_classifier[example2].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_answer_relevance_classifier/test_run_classifier[example3].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_answer_relevance_classifier/test_run_classifier[example3].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_answer_relevance_composite/test_run_composite[example1].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_answer_relevance_composite/test_run_composite[example1].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_answer_relevance_composite/test_run_composite[example2].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_answer_relevance_composite/test_run_composite[example2].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_answer_relevance_composite/test_run_composite[example3].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_answer_relevance_composite/test_run_composite[example3].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_answer_relevance_rewriter/test_run_rewriter[example1].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_answer_relevance_rewriter/test_run_rewriter[example1].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_answer_relevance_rewriter/test_run_rewriter[example2].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_answer_relevance_rewriter/test_run_rewriter[example2].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_answerability/test_run_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_answerability/test_run_model.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_certainty/test_run_composite.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_certainty/test_run_composite.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_certainty/test_run_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_certainty/test_run_model.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_citations/test_run_composite.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_citations/test_run_composite.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_citations/test_run_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_citations/test_run_model.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_context_relevancy/test_run_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_context_relevancy/test_run_model.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_granite_3_2/test_run_processor[backend_litellm-simple].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_granite_3_2/test_run_processor[backend_litellm-simple].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_granite_3_2/test_run_processor[backend_litellm-thinking_tag].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_granite_3_2/test_run_processor[backend_litellm-thinking_tag].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_granite_3_2/test_run_processor[backend_openai-simple].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_granite_3_2/test_run_processor[backend_openai-simple].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_granite_3_2/test_run_processor[backend_openai-thinking_tag].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_granite_3_2/test_run_processor[backend_openai-thinking_tag].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_hallucinations/test_run_composite.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_hallucinations/test_run_composite.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_hallucinations/test_run_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_hallucinations/test_run_model.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_hallucinations/test_run_processor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_hallucinations/test_run_processor.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_litellm-simple].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_litellm-simple].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_litellm-stop_strings].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_litellm-stop_strings].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_litellm-thinking_tag].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_litellm-thinking_tag].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_openai-simple].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_openai-simple].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_openai-stop_strings].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_openai-stop_strings].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_openai-thinking_tag].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_make_io_processor/test_make_io_processor[backend_openai-thinking_tag].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_prm/test_run_composite.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_prm/test_run_composite.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_prm/test_run_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_prm/test_run_model.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_query_expansion/test_run_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_query_expansion/test_run_model.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_query_rewrite/test_request_processor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_query_rewrite/test_request_processor.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_query_rewrite/test_run_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_query_rewrite/test_run_model.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_rerank/test_rerank_request_processor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_rerank/test_rerank_request_processor.yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_voting/test_mbrd_majority_voting[backend_openai].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_voting/test_mbrd_majority_voting[backend_openai].yaml -------------------------------------------------------------------------------- /tests/io/cassettes/test_voting/test_numeric_voting[backend_openai].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/cassettes/test_voting/test_numeric_voting[backend_openai].yaml -------------------------------------------------------------------------------- /tests/io/data/test_answer_relevance3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/data/test_answer_relevance3.yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_completion_presence_param[backend_litellm].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_completion_presence_param[backend_litellm].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_completion_presence_param[backend_openai].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_completion_presence_param[backend_openai].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_completion_repetition_param[backend_litellm].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_completion_repetition_param[backend_litellm].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_completion_repetition_param[backend_openai].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_completion_repetition_param[backend_openai].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_multiple_return[backend_openai-custom_system_prompt].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_multiple_return[backend_openai-custom_system_prompt].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_multiple_return[backend_openai-functions].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_multiple_return[backend_openai-functions].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_multiple_return[backend_openai-simple].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_multiple_return[backend_openai-simple].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_multiple_return[backend_openai-stop_strings].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_multiple_return[backend_openai-stop_strings].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_multiple_return[backend_openai-thinking_tag].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_multiple_return[backend_openai-thinking_tag].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_litellm-custom_system_prompt].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_litellm-custom_system_prompt].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_litellm-functions].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_litellm-functions].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_litellm-simple].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_litellm-simple].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_litellm-stop_strings].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_litellm-stop_strings].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_litellm-thinking_tag].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_litellm-thinking_tag].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_openai-custom_system_prompt].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_openai-custom_system_prompt].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_openai-functions].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_openai-functions].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_openai-simple].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_openai-simple].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_openai-stop_strings].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_openai-stop_strings].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_openai-thinking_tag].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/cassettes/test_granite_3_2/test_run_processor[backend_openai-thinking_tag].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_2/input_processors/test_granite_3_2_input_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/input_processors/test_granite_3_2_input_processor.py -------------------------------------------------------------------------------- /tests/io/granite_3_2/input_processors/testdata/test_reasoning_prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/input_processors/testdata/test_reasoning_prompt.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/input_processors/testdata/test_remove_special_tokens_expected_prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/input_processors/testdata/test_remove_special_tokens_expected_prompt.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/input_processors/testdata/test_remove_special_tokens_input_json.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/input_processors/testdata/test_remove_special_tokens_input_json.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/test_granite_3_2_output_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/test_granite_3_2_output_parser.py -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/test_granite_3_2_output_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/test_granite_3_2_output_processor.py -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/testdata/test_document_source.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/testdata/test_document_source.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/testdata/test_output.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/testdata/test_output.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/testdata/test_output_with_citation.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/testdata/test_output_with_citation.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/testdata/test_output_with_citation_from_source.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/testdata/test_output_with_citation_from_source.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/testdata/test_output_with_citation_hallucinations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/testdata/test_output_with_citation_hallucinations.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/testdata/test_output_with_colons_citation_text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/testdata/test_output_with_colons_citation_text.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/testdata/test_output_with_invalid_citation.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/testdata/test_output_with_invalid_citation.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/testdata/test_output_with_multiple_citations_per_document.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/testdata/test_output_with_multiple_citations_per_document.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/testdata/test_raw_reasoning_output.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/testdata/test_raw_reasoning_output.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/testdata/test_reasoning_output_processor_response.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/testdata/test_reasoning_output_processor_response.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/output_processors/testdata/test_reasoning_output_processor_thought.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/output_processors/testdata/test_reasoning_output_processor_thought.txt -------------------------------------------------------------------------------- /tests/io/granite_3_2/test_granite_3_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_2/test_granite_3_2.py -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_completion_presence_param[backend_3_3_openai].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_completion_presence_param[backend_3_3_openai].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_completion_repetition_param[backend_3_3_litellm].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_completion_repetition_param[backend_3_3_litellm].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-custom_system_prompt].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-custom_system_prompt].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-documents].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-documents].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-functions].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-functions].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-simple].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-simple].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-stop_strings].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-stop_strings].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-thinking_tag].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_multiple_return[backend_3_3_openai-thinking_tag].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-custom_system_prompt].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-custom_system_prompt].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-documents].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-documents].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-functions].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-functions].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-simple].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-simple].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-stop_strings].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-stop_strings].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-thinking_tag].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_litellm-thinking_tag].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-custom_system_prompt].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-custom_system_prompt].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-documents].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-documents].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-functions].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-functions].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-simple].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-simple].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-stop_strings].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-stop_strings].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-thinking_tag].yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/cassettes/test_granite_3_3/test_run_processor[backend_3_3_openai-thinking_tag].yaml -------------------------------------------------------------------------------- /tests/io/granite_3_3/input_processors/test_granite_3_3_input_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/input_processors/test_granite_3_3_input_processor.py -------------------------------------------------------------------------------- /tests/io/granite_3_3/input_processors/testdata/test_reasoning_prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/input_processors/testdata/test_reasoning_prompt.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/input_processors/testdata/test_remove_special_tokens_expected_prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/input_processors/testdata/test_remove_special_tokens_expected_prompt.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/input_processors/testdata/test_remove_special_tokens_input_json.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/input_processors/testdata/test_remove_special_tokens_input_json.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/test_granite_3_3_output_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/test_granite_3_3_output_parser.py -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/test_granite_3_3_output_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/test_granite_3_3_output_processor.py -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/testdata/test_document_source.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/testdata/test_document_source.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/testdata/test_output.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/testdata/test_output.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/testdata/test_output_with_citation.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/testdata/test_output_with_citation.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/testdata/test_output_with_citation_from_source.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/testdata/test_output_with_citation_from_source.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/testdata/test_output_with_citation_hallucinations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/testdata/test_output_with_citation_hallucinations.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/testdata/test_output_with_colons_citation_text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/testdata/test_output_with_colons_citation_text.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/testdata/test_output_with_invalid_citation.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/testdata/test_output_with_invalid_citation.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/testdata/test_output_with_multiple_citations_per_document.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/testdata/test_output_with_multiple_citations_per_document.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/testdata/test_raw_reasoning_output.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/testdata/test_raw_reasoning_output.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/testdata/test_reasoning_output_processor_response.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/testdata/test_reasoning_output_processor_response.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/output_processors/testdata/test_reasoning_output_processor_thought.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/output_processors/testdata/test_reasoning_output_processor_thought.txt -------------------------------------------------------------------------------- /tests/io/granite_3_3/test_granite_3_3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/granite_3_3/test_granite_3_3.py -------------------------------------------------------------------------------- /tests/io/test_answer_relevance_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_answer_relevance_classifier.py -------------------------------------------------------------------------------- /tests/io/test_answer_relevance_composite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_answer_relevance_composite.py -------------------------------------------------------------------------------- /tests/io/test_answer_relevance_rewriter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_answer_relevance_rewriter.py -------------------------------------------------------------------------------- /tests/io/test_answerability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_answerability.py -------------------------------------------------------------------------------- /tests/io/test_certainty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_certainty.py -------------------------------------------------------------------------------- /tests/io/test_citations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_citations.py -------------------------------------------------------------------------------- /tests/io/test_context_relevancy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_context_relevancy.py -------------------------------------------------------------------------------- /tests/io/test_hallucinations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_hallucinations.py -------------------------------------------------------------------------------- /tests/io/test_hyde_rewrite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_hyde_rewrite.py -------------------------------------------------------------------------------- /tests/io/test_make_io_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_make_io_processor.py -------------------------------------------------------------------------------- /tests/io/test_prm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_prm.py -------------------------------------------------------------------------------- /tests/io/test_query_expansion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_query_expansion.py -------------------------------------------------------------------------------- /tests/io/test_query_rewrite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_query_rewrite.py -------------------------------------------------------------------------------- /tests/io/test_rerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_rerank.py -------------------------------------------------------------------------------- /tests/io/test_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_retrieval.py -------------------------------------------------------------------------------- /tests/io/test_voting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/io/test_voting.py -------------------------------------------------------------------------------- /tests/test_backends.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/test_backends.py -------------------------------------------------------------------------------- /tests/test_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/test_factory.py -------------------------------------------------------------------------------- /tests/test_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/test_types.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tests/test_utils.py -------------------------------------------------------------------------------- /tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ibm-granite/granite-io/HEAD/tox.ini --------------------------------------------------------------------------------