local_rqa namespace

Subpackages

• local_rqa.evaluation package
  • Submodules
  • local_rqa.evaluation.evaluator module
  • local_rqa.evaluation.metrics module
  • local_rqa.evaluation.scores module
  • local_rqa.evaluation.utils module
    • normalize_answer()
  • Module contents
• local_rqa.guardrails package
  • Submodules
  • local_rqa.guardrails.base module
  • Module contents
• local_rqa.pipelines package
  • Submodules
  • local_rqa.pipelines.base module
  • local_rqa.pipelines.prompts module
  • local_rqa.pipelines.retrieval_qa module
  • Module contents
• local_rqa.qa_llms package
  • Submodules
  • local_rqa.qa_llms.base module
  • local_rqa.qa_llms.fid module
  • local_rqa.qa_llms.huggingface module
  • local_rqa.qa_llms.openai module
  • local_rqa.qa_llms.prompts module
  • local_rqa.qa_llms.sglang module
  • local_rqa.qa_llms.tgi module
  • local_rqa.qa_llms.vllm module
  • Module contents
• local_rqa.retrievers package
  • Submodules
  • local_rqa.retrievers.base module
  • local_rqa.retrievers.bm25_retriever module
  • local_rqa.retrievers.faiss_retriever module
  • Module contents
• local_rqa.schema package
  • Submodules
  • local_rqa.schema.dialogue module
  • local_rqa.schema.document module
  • Module contents
• local_rqa.serve package
  • Submodules
  • local_rqa.serve.base_model_worker module
  • local_rqa.serve.controller module
  • local_rqa.serve.gradio_dialogue module
  • local_rqa.serve.gradio_rqa module
  • local_rqa.serve.gradio_static_server module
  • local_rqa.serve.gradio_web_server module
  • local_rqa.serve.model_worker module
  • local_rqa.serve.test_message module
  • Module contents
• local_rqa.text_loaders package
  • Submodules
  • local_rqa.text_loaders.base module
  • local_rqa.text_loaders.langchain_text_loader module
  • local_rqa.text_loaders.llamaindex_text_loader module
  • Module contents
• local_rqa.trainers namespace
  • Subpackages
    • local_rqa.trainers.qa_llm namespace
      • Submodules
      • local_rqa.trainers.qa_llm.arguments module
      • local_rqa.trainers.qa_llm.datasets module
      • local_rqa.trainers.qa_llm.supervised_fid_trainer module
      • local_rqa.trainers.qa_llm.supervised_trainer module
    • local_rqa.trainers.retriever namespace
      • Submodules
      • local_rqa.trainers.retriever.arguments module
      • local_rqa.trainers.retriever.datasets module
      • local_rqa.trainers.retriever.embeddings module
      • local_rqa.trainers.retriever.retriever_fid_trainer module
      • local_rqa.trainers.retriever.retriever_replug_trainer module
      • local_rqa.trainers.retriever.retriever_trainer module
  • Submodules
  • local_rqa.trainers.dist_utils module
  • local_rqa.trainers.utils module

Submodules

local_rqa.base module

class local_rqa.base.Component

Bases: ABC

building block of a pipeline, which will call the run method iteratively

abstract run(*args, **kwargs)

main entrypoint for the component. keywords in run_input_keys will be passed in as arguments for this method

run_input_keys = []

local_rqa.constants module

class local_rqa.constants.AccelerationFramework(value)

Bases: Enum

An enumeration.

SGLANG = 'sglang::'

TGI = 'tgi::'

VLLM = 'vllm::'

class local_rqa.constants.ErrorCode(value)

Bases: IntEnum

https://platform.openai.com/docs/guides/error-codes/api-errors

CONTEXT_OVERFLOW = 40303

CONTROLLER_NO_WORKER = 50005

CONTROLLER_WORKER_TIMEOUT = 50006

CUDA_OUT_OF_MEMORY = 50002

ENGINE_OVERLOADED = 42903

GRADIO_REQUEST_ERROR = 50003

GRADIO_STREAM_UNKNOWN_ERROR = 50004

INCORRECT_AUTH_KEY = 40102

INTERNAL_ERROR = 50001

INVALID_AUTH_KEY = 40101

INVALID_MODEL = 40301

NO_PERMISSION = 40103

PARAM_OUT_OF_RANGE = 40302

QUOTA_EXCEEDED = 42902

RATE_LIMIT = 42901

VALIDATION_TYPE_ERROR = 40001

local_rqa.utils module