Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Index your code
hub
/
github.com/adithya-s-k/VARAG
/ functions
Functions
240 in github.com/adithya-s-k/VARAG
⨍
Functions
240
◇
Types & classes
23
↓ 28 callers
Method
query
Abstract method for querying the model
varag/llms/base.py:11
↓ 28 callers
Method
search
(self, query: str, k: int = 5)
varag/rag/_simpleRAG.py:211
↓ 18 callers
Method
change_table
(self, new_table_name: str)
varag/rag/_simpleRAG.py:399
↓ 18 callers
Method
index
( self, path: Union[str, List[str]], recursive: bool = False, chunking_strateg
varag/rag/_simpleRAG.py:78
↓ 8 callers
Function
get_model_colpali
(model_name: str = "vidore/colpali-v1.3")
varag/utils.py:14
↓ 4 callers
Method
chunk
(self, text: str)
varag/chunking/__init__.py:8
↓ 4 callers
Function
query_data_single
(query, retrieved_results, model_name)
demo.py:492
↓ 4 callers
Function
query_data_single
(query, retrieved_results, model_name)
examples/inference_colpali/demo_with_heatmaps.py:715
↓ 4 callers
Function
query_data_single
Helper function to query a single RAG model with error handling
examples/inference_colpali/modal_demo_heatmaps_comparing_colpali_models.py:329
↓ 3 callers
Function
_gen
()
examples/base_implementation/hybrid_colpali_embed.py:200
↓ 3 callers
Function
_gen
()
examples/base_implementation/colpali_embed.py:185
↓ 3 callers
Function
analyze_multiple_images
Analysis of multiple images with the same query - processes ALL tokens. Args: similarity_mapper: ColPaliSimilarityMapper instanc
varag/utils.py:294
↓ 3 callers
Function
create_similarity_mapper
Factory function to create a ColPaliSimilarityMapper instance. Args: model: ColPali model instance processor: ColPali pr
varag/utils.py:270
↓ 2 callers
Method
_encode_image
Encode a PIL Image to base64.
varag/vlms/provider/litellm.py:71
↓ 2 callers
Method
_join_docs
(self, docs: List[str], separator: str)
examples/base_implementation/text_embed.py:127
↓ 2 callers
Method
_process_file
Process a single file for text extraction and embedding. Args: file_path: Path to the file to process
varag/rag/_simpleRAG.py:120
↓ 2 callers
Function
embedder
(chunk)
examples/base_implementation/text_embed.py:150
↓ 2 callers
Method
generate_aggregated_similarity_map
Generate aggregated similarity map by averaging across all query tokens.
examples/inference_colpali/modal_demo_heatmaps_comparing_colpali_models.py:502
↓ 2 callers
Method
process_file
( self, file_path: str, metadata: Dict[str, str] = None, verbose: bool = False )
varag/rag/_colpaliRAG.py:187
↓ 2 callers
Method
process_file
( self, file_path: str, metadata: Dict[str, str] = None, verbose: bool = False )
varag/rag/_hybridColpaliRAG.py:184
↓ 2 callers
Method
process_file
( self, file_path: str, metadata: Dict[str, str] = None, verbose: bool = False )
varag/rag/_visionRAG.py:103
↓ 1 callers
Method
__init__
( self, chunk_size: int = 4000, chunk_overlap: int = 200, length_function: Cal
examples/base_implementation/text_embed.py:90
↓ 1 callers
Method
_initialize_client
(self)
varag/vlms/provider/openai.py:50
↓ 1 callers
Method
_initialize_client
(self)
varag/llms/provider/openai.py:50
↓ 1 callers
Method
_prepare_image_content
(self, image: Union[str, Image.Image])
varag/vlms/provider/openai.py:53
↓ 1 callers
Method
_prepare_messages
Prepare messages for the VLM API call. Args: query: The user query images: List of images (paths or
varag/vlms/provider/litellm.py:77
↓ 1 callers
Method
_prepare_messages
Prepare messages for the LLM API call. Args: query: The user query system_prompt: Optional system pr
varag/llms/provider/litellm.py:73
↓ 1 callers
Method
_validate_model
(self, model: str)
varag/vlms/provider/openai.py:40
↓ 1 callers
Method
_validate_model
(self, model: str)
varag/llms/provider/openai.py:40
↓ 1 callers
Method
_validate_setup
Validate the environment setup and model access.
varag/vlms/provider/litellm.py:60
↓ 1 callers
Method
_validate_setup
Validate the environment setup and model access.
varag/llms/provider/litellm.py:62
↓ 1 callers
Method
analyze_image_with_query
Complete analysis pipeline for a single image - processes ALL tokens. Args: image_input: Base64-encoded image st
varag/utils.py:192
↓ 1 callers
Function
base64_to_pil
(base64_str: str)
examples/base_implementation/hybrid_colpali_embed.py:85
↓ 1 callers
Function
base64_to_pil
(base64_str: str)
examples/base_implementation/colpali_embed.py:70
↓ 1 callers
Function
base64_to_pil
Convert base64 string to PIL Image
examples/inference_colpali/modal_demo_heatmaps_comparing_colpali_models.py:487
↓ 1 callers
Method
base64_to_pil
Convert base64 string to PIL Image
varag/utils.py:66
↓ 1 callers
Method
base64_to_pil
(self, base64_str: str)
varag/rag/_colpaliRAG.py:184
↓ 1 callers
Method
base64_to_pil
(self, base64_str: str)
varag/rag/_hybridColpaliRAG.py:181
↓ 1 callers
Function
bot_streaming
(message, history, pdf_file, max_new_tokens=250)
examples/base_implementation/colpali_simple.py:66
↓ 1 callers
Method
calculate_ndcg
Calculate NDCG score for retrieved results.
varag/rag/_simpleRAG.py:291
↓ 1 callers
Function
compare_colpali_models
Compare both ColPali models with BOTH aggregated heatmaps AND individual token analysis
examples/inference_colpali/modal_demo_heatmaps_comparing_colpali_models.py:679
↓ 1 callers
Function
create_aggregated_comparison_heatmaps
Create side-by-side comparison of aggregated heatmaps.
examples/inference_colpali/modal_demo_heatmaps_comparing_colpali_models.py:610
↓ 1 callers
Method
create_all_similarity_visualizations
Create similarity visualizations for all tokens using plot_all_similarity_maps. Args: image: Original PIL Image
varag/utils.py:140
↓ 1 callers
Function
create_db
(docs_storage, table_name: str = "demo", db_path: str = "lancedb")
examples/base_implementation/hybrid_colpali_embed.py:199
↓ 1 callers
Function
create_db
(docs_storage, table_name: str = "demo", db_path: str = "lancedb")
examples/base_implementation/colpali_embed.py:184
↓ 1 callers
Function
create_gradio_interface
()
examples/hybridColpaliDemo.py:94
↓ 1 callers
Function
create_gradio_interface
()
examples/colpaliDemo.py:83
↓ 1 callers
Function
create_gradio_interface
()
examples/textDemo.py:70
↓ 1 callers
Function
embed_images
(images: List[Image.Image])
examples/base_implementation/hybrid_colpali_embed.py:58
↓ 1 callers
Function
embed_images
(images: List[Image.Image])
examples/base_implementation/jina_clip_embed.py:75
↓ 1 callers
Function
embed_images
(images: List[Image.Image])
examples/base_implementation/clip_embed.py:69
↓ 1 callers
Method
embed_images
(self, images: List[Image.Image], verbose: bool = False)
varag/rag/_hybridColpaliRAG.py:100
↓ 1 callers
Method
embed_images
(self, images: List[Image.Image], verbose: bool = False)
varag/rag/_visionRAG.py:94
↓ 1 callers
Method
embed_images_colpali
(self, images: List[Image.Image], verbose: bool = False)
varag/rag/_colpaliRAG.py:147
↓ 1 callers
Method
embed_images_colpali
(self, images: List[Image.Image], verbose: bool = False)
varag/rag/_hybridColpaliRAG.py:144
↓ 1 callers
Function
embed_text
(text: str)
examples/base_implementation/hybrid_colpali_embed.py:63
↓ 1 callers
Function
embed_text
(text: str)
examples/base_implementation/jina_clip_embed.py:112
↓ 1 callers
Function
embed_text
(text: str)
examples/base_implementation/clip_embed.py:102
↓ 1 callers
Method
embed_text
(self, text: str)
varag/rag/_hybridColpaliRAG.py:337
↓ 1 callers
Function
embedd_docs
Returns embeddings for all pages in a single PDF document. Args: pdf_path (str): Path to the PDF document. model (ColPali):
examples/base_implementation/hybrid_colpali_embed.py:178
↓ 1 callers
Function
embedd_docs
Returns embeddings for all pages in a single PDF document. Args: pdf_path (str): Path to the PDF document. model (ColPali):
examples/base_implementation/colpali_embed.py:163
↓ 1 callers
Method
extract_text_from_pdf
(self, pdf_path: str)
varag/rag/_colpaliRAG.py:229
↓ 1 callers
Method
extract_text_from_pdf
(self, pdf_path: str)
varag/rag/_hybridColpaliRAG.py:226
↓ 1 callers
Function
extract_text_with_pymupdf
(pdf_path)
examples/base_implementation/text_embed.py:24
↓ 1 callers
Method
extract_text_with_pymupdf
(self, pdf_path)
varag/rag/_simpleRAG.py:71
↓ 1 callers
Function
flatten_and_zero_pad
Flattens a PyTorch tensor and zero-pads it to a desired length. Args: tensor: The input PyTorch tensor. desired_length: The desir
examples/base_implementation/hybrid_colpali_embed.py:239
↓ 1 callers
Function
flatten_and_zero_pad
Flattens a PyTorch tensor and zero-pads it to a desired length. Args: tensor: The input PyTorch tensor. desired_length: The desir
examples/base_implementation/colpali_embed.py:221
↓ 1 callers
Method
generate_similarity_maps
Generate similarity maps for ALL tokens in the query. Args: image: PIL Image to analyze query: Query
varag/utils.py:76
↓ 1 callers
Function
get_base64_image
(image: Union[str, Image.Image])
examples/base_implementation/hybrid_colpali_embed.py:93
↓ 1 callers
Function
get_base64_image
(image: Union[str, Image.Image])
examples/base_implementation/colpali_embed.py:78
↓ 1 callers
Function
get_model_colpali
(base_model_id: Optional[str] = None)
examples/base_implementation/hybrid_colpali_embed.py:68
↓ 1 callers
Function
get_model_colpali
(base_model_id: Optional[str] = None)
examples/base_implementation/colpali_embed.py:53
↓ 1 callers
Function
get_pdf_embedding
(pdf_path: str, model, processor)
examples/base_implementation/hybrid_colpali_embed.py:129
↓ 1 callers
Function
get_pdf_embedding
(pdf_path: str, model, processor)
examples/base_implementation/colpali_embed.py:114
↓ 1 callers
Function
get_pdf_images
(pdf_path)
examples/base_implementation/hybrid_colpali_embed.py:116
↓ 1 callers
Function
get_pdf_images
(pdf_path)
examples/base_implementation/colpali_embed.py:101
↓ 1 callers
Function
get_query_embedding
(query: str, model, processor)
examples/base_implementation/hybrid_colpali_embed.py:158
↓ 1 callers
Function
get_query_embedding
(query: str, model, processor)
examples/base_implementation/colpali_embed.py:143
↓ 1 callers
Method
get_query_embedding
(self, query: str, model, processor)
varag/rag/_colpaliRAG.py:321
↓ 1 callers
Method
get_query_embedding
(self, query: str, model, processor)
varag/rag/_hybridColpaliRAG.py:319
↓ 1 callers
Function
gradio_interface
()
demo.py:301
↓ 1 callers
Function
gradio_interface
()
examples/visionDemo.py:67
↓ 1 callers
Function
gradio_interface
()
examples/base_implementation/text_embed.py:222
↓ 1 callers
Function
gradio_interface
()
examples/base_implementation/jina_clip_embed.py:145
↓ 1 callers
Function
gradio_interface
()
examples/base_implementation/clip_embed.py:137
↓ 1 callers
Function
gradio_interface
()
examples/inference_colpali/demo_with_heatmaps.py:329
↓ 1 callers
Method
index
( self, data_path: Union[str, List[str]], overwrite: bool = False, recursive:
varag/rag/_visionRAG.py:134
↓ 1 callers
Function
ingest_pdfs
(pdf_paths: List[str], use_docling: bool = False)
examples/base_implementation/text_embed.py:32
↓ 1 callers
Function
lanceDBConnection
(chunks, embeddings)
examples/base_implementation/text_embed.py:171
↓ 1 callers
Function
parse_args
()
demo.py:577
↓ 1 callers
Function
parse_args
()
examples/hybridColpaliDemo.py:148
↓ 1 callers
Function
parse_args
()
examples/colpaliDemo.py:131
↓ 1 callers
Function
parse_args
()
examples/textDemo.py:108
↓ 1 callers
Function
parse_args
()
examples/visionDemo.py:112
↓ 1 callers
Function
parse_args
()
examples/inference_colpali/demo_with_heatmaps.py:816
↓ 1 callers
Function
pdf_to_images
(pdf_path: str)
examples/base_implementation/jina_clip_embed.py:47
↓ 1 callers
Function
pdf_to_images
(pdf_path: str)
examples/base_implementation/clip_embed.py:47
↓ 1 callers
Method
pdf_to_images
(self, pdf_path: str, verbose: bool = False)
varag/rag/_colpaliRAG.py:119
↓ 1 callers
Method
pdf_to_images
(self, pdf_path: str, verbose: bool = False)
varag/rag/_hybridColpaliRAG.py:116
↓ 1 callers
Method
pdf_to_images
(self, pdf_path: str, verbose: bool = False)
varag/rag/_visionRAG.py:66
next →
1–100 of 240, ranked by callers