Functions — github.com/datawhalechina/all-in-rag

hub / github.com/datawhalechina/all-in-rag / functions

Functions495 in github.com/datawhalechina/all-in-rag

⨍ Functions 495◇ Types & classes 94

↓ 1 callersMethodconnect_database

连接SQLite数据库

code/C4/text2sql/text2sql_agent.py:23

↓ 1 callersMethodconnect_milvus

连接Milvus数据库

code/C4/03_text2sql_demo_v2.py:59

↓ 1 callersFunctioncontinue_processing

继续处理中断的任务

code/C9/agent(代码系ai生成)/batch_manager.py:124

↓ 1 callersFunctionconvert_to_custom_text_state_dict

(state_dict: dict)

code/C3/visual_bge/visual_bge/eva_clip/model.py:360

↓ 1 callersFunctionconvert_weights_to_lp

Convert applicable model parameters to low-precision (bf16 or fp16)

code/C3/visual_bge/visual_bge/eva_clip/model.py:328

↓ 1 callersMethodcreate_collection

创建Milvus集合 Args: force_recreate: 是否强制重新创建集合 Returns: 是否创建成功

code/C9/rag_modules/milvus_index_construction.py:122

↓ 1 callersFunctioncreate_demo_database

创建演示数据库

code/C4/03_text2sql_demo.py:47

↓ 1 callersMethodcreate_entity_key_values

为实体创建键值对结构每个实体使用其名称作为唯一索引键

code/C9/rag_modules/graph_indexing.py:60

↓ 1 callersFunctioncreate_eva_vision_and_transforms

( model_name: str, pretrained: Optional[str] = None, precision: str = 'fp32',

code/C3/visual_bge/visual_bge/eva_clip/factory.py:416

↓ 1 callersMethodcreate_relation_key_values

为关系创建键值对结构关系可能有多个索引键，包含从LLM增强的全局主题

code/C9/rag_modules/graph_indexing.py:169

↓ 1 callersMethoddecode

(self, tokens)

code/C3/visual_bge/visual_bge/eva_clip/tokenizer.py:147

↓ 1 callersMethoddeduplicate_entities_and_relations

去重相同的实体和关系，优化图操作

code/C9/rag_modules/graph_indexing.py:295

↓ 1 callersFunctiondefault_bpe

()

code/C3/visual_bge/visual_bge/eva_clip/tokenizer.py:21

↓ 1 callersMethoddelete_collection

删除集合 Returns: 是否删除成功

code/C9/rag_modules/milvus_index_construction.py:440

↓ 1 callersFunctiondemo

简单演示

code/C4/03_text2sql_demo_v2.py:323

↓ 1 callersFunctiondemo_normalization

演示标准化功能

code/C9/agent(代码系ai生成)/amount_normalizer.py:204

↓ 1 callersFunctiondownload_visualized_bge_model

下载 Visual BGE 模型权重文件如果模型文件不存在，则从 Hugging Face 下载

code/C3/download_model.py:4

↓ 1 callersMethoddual_level_retrieval

双层检索：结合实体级和主题级检索

code/C9/rag_modules/hybrid_retrieval.py:470

↓ 1 callersMethodencode

(self, text)

code/C3/visual_bge/visual_bge/eva_clip/tokenizer.py:139

↓ 1 callersMethodencode_image

(self, image, normalize: bool = False)

code/C3/visual_bge/visual_bge/eva_clip/model.py:246

↓ 1 callersMethodencode_image

(self, image, normalize: bool = False)

code/C3/visual_bge/visual_bge/eva_clip/model.py:308

↓ 1 callersMethodencode_query

(self, image_path: str, text: str)

code/C3/04_multi_milvus.py:25

↓ 1 callersMethodencode_text

(self, text, normalize: bool = False)

code/C3/visual_bge/visual_bge/eva_clip/model.py:250

↓ 1 callersMethodencode_text

(self, text, normalize: bool = False)

code/C3/visual_bge/visual_bge/eva_clip/model.py:312

↓ 1 callersMethodencode_text_dense

使用SigLIP编码文本为密集向量

code/C4/01_hybrid_search_v2.py:66

↓ 1 callersMethodencode_text_sparse

使用TF-IDF编码文本为稀疏向量

code/C4/01_hybrid_search_v2.py:89

↓ 1 callersMethodentity_level_retrieval

实体级检索：专注于具体实体和关系使用图索引的键值对结构进行检索

code/C9/rag_modules/hybrid_retrieval.py:222

↓ 1 callersMethodexplain_routing_decision

解释路由决策过程

code/C9/rag_modules/intelligent_query_router.py:284

↓ 1 callersMethodexport_to_rf2_format

导出为RF2标准格式

code/C9/agent(代码系ai生成)/recipe_ai_agent.py:1061

↓ 1 callersMethodextract_knowledge_subgraph

提取知识子图：获取实体相关的完整知识网络这体现了图RAG的整体性思维

code/C9/rag_modules/graph_rag_retrieval.py:350

↓ 1 callersMethodextract_query_keywords

提取查询关键词：实体级 + 主题级

code/C9/rag_modules/hybrid_retrieval.py:163

↓ 1 callersMethodfit_sparse

拟合稀疏向量模型（TF-IDF）

code/C4/01_hybrid_search_v2.py:59

↓ 1 callersMethodfix_init_weight

(self)

code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:443

↓ 1 callersMethodfix_sql

修复SQL语句

code/C4/text2sql/sql_generator.py:52

↓ 1 callersFunctiongather_features

( image_features, text_features, local_loss=False, gather_with_grad=False,

code/C3/visual_bge/visual_bge/eva_clip/loss.py:21

↓ 1 callersMethodgenerate_adaptive_answer_stream

LightRAG风格的流式答案生成（带重试机制）

code/C9/rag_modules/generation_integration.py:91

↓ 1 callersMethodgenerate_basic_answer_stream

生成基础回答 - 流式输出 Args: query: 用户查询 context_docs: 上下文文档列表 Yields: 生成的回答片段

code/C8/rag_modules/generation_integration.py:277

↓ 1 callersMethodgenerate_list_answer

生成列表式回答 - 适用于推荐类查询 Args: query: 用户查询 context_docs: 上下文文档列表 Returns: 列表式回答

code/C8/rag_modules/generation_integration.py:248

↓ 1 callersMethodgenerate_sql

生成SQL语句

code/C4/text2sql/sql_generator.py:17

↓ 1 callersMethodgenerate_step_by_step_answer

生成分步骤回答 Args: query: 用户查询 context_docs: 上下文文档列表 Returns: 分步骤的详细回答

code/C8/rag_modules/generation_integration.py:88

↓ 1 callersMethodgenerate_step_by_step_answer_stream

生成详细步骤回答 - 流式输出 Args: query: 用户查询 context_docs: 上下文文档列表 Yields: 详细步骤回答片段

code/C8/rag_modules/generation_integration.py:312

↓ 1 callersFunctiongeometric_progression

(a, r, n)

code/C3/visual_bge/visual_bge/eva_clip/utils.py:162

↓ 1 callersMethodget_collection_stats

获取集合统计信息 Returns: 统计信息字典

code/C9/rag_modules/milvus_index_construction.py:417

↓ 1 callersFunctionget_recipe_directory

获取菜谱目录

code/C9/agent(代码系ai生成)/run_ai_agent.py:41

↓ 1 callersMethodget_relations_by_key

根据索引键获取关系

code/C9/rag_modules/graph_indexing.py:368

↓ 1 callersMethodget_statistics

获取数据统计信息 Returns: 统计信息字典

code/C8/rag_modules/data_preparation.py:260

↓ 1 callersMethodget_supported_categories

对外提供支持的分类标签列表

code/C8/rag_modules/data_preparation.py:130

↓ 1 callersMethodget_supported_difficulties

对外提供支持的难度标签列表

code/C8/rag_modules/data_preparation.py:135

↓ 1 callersMethodgradient_checkpointing_enable

(self, **kwargs)

code/C3/visual_bge/visual_bge/modeling.py:108

↓ 1 callersMethodgraph_structure_reasoning

基于图结构的推理：这是图RAG的智能之处不仅检索信息，还能进行逻辑推理

code/C9/rag_modules/graph_rag_retrieval.py:408

↓ 1 callersMethodimg_token_embedding

(self, images)

code/C3/visual_bge/visual_bge/modeling.py:301

↓ 1 callersMethodinfer_category_from_path

根据文件路径推断菜谱分类

code/C9/agent(代码系ai生成)/recipe_ai_agent.py:124

↓ 1 callersMethodinit_parameters

(self)

code/C3/visual_bge/visual_bge/eva_clip/modified_resnet.py:141

↓ 1 callersMethodinit_parameters

(self)

code/C3/visual_bge/visual_bge/eva_clip/transformer.py:686

↓ 1 callersMethodinitialize_system

初始化所有模块

code/C8/main.py:57

↓ 1 callersMethodinitialize_system

初始化高级图RAG系统

code/C9/main.py:61

↓ 1 callersMethodinsert_data

插入数据

code/C3/work_hybrid_multimodal_search.py:214

↓ 1 callersFunctionis_global_master

(args)

code/C3/visual_bge/visual_bge/eva_clip/utils.py:293

↓ 1 callersFunctionis_local_master

(args)

code/C3/visual_bge/visual_bge/eva_clip/utils.py:296

↓ 1 callersFunctionis_pretrained_cfg

(model: str, tag: str)

code/C3/visual_bge/visual_bge/eva_clip/pretrained.py:221

↓ 1 callersFunctionlist_models

enumerate available model architectures based on config files

code/C3/visual_bge/visual_bge/eva_clip/factory.py:53

↓ 1 callersFunctionlist_openai_models

Returns the names of available CLIP models

code/C3/visual_bge/visual_bge/eva_clip/openai.py:18

↓ 1 callersFunctionlist_pretrained_models_by_tag

return all models having the specified pretrain tag

code/C3/visual_bge/visual_bge/eva_clip/pretrained.py:203

↓ 1 callersFunctionlist_pretrained_tags_by_model

return all pretrain tags for the specified model architecture

code/C3/visual_bge/visual_bge/eva_clip/pretrained.py:213

↓ 1 callersFunctionload_checkpoint

(model, checkpoint_path, model_key="model|module|state_dict", strict=True)

code/C3/visual_bge/visual_bge/eva_clip/factory.py:110

↓ 1 callersMethodload_data

加载所有知识库数据

code/C4/text2sql/knowledge_base.py:53

↓ 1 callersMethodload_index

从配置的路径加载向量索引 Returns: 加载的向量存储对象，如果加载失败返回None

code/C8/rag_modules/index_construction.py:95

↓ 1 callersMethodload_knowledge_base

加载知识库

code/C4/text2sql/text2sql_agent.py:34

↓ 1 callersMethodload_model

(self, model_weight)

code/C3/visual_bge/visual_bge/modeling.py:105

↓ 1 callersFunctionload_openai_model

Load a CLIP model Parameters ---------- name : str A model name listed by `clip.available_models()`, or the path to a model check

code/C3/visual_bge/visual_bge/eva_clip/openai.py:23

↓ 1 callersFunctionload_pretrained_checkpoint

( model, visual_checkpoint_path, text_checkpoint_path, strict=True, vi

code/C3/visual_bge/visual_bge/eva_clip/factory.py:163

↓ 1 callersMethodload_progress

加载处理进度

code/C9/agent(代码系ai生成)/recipe_ai_agent.py:508

↓ 1 callersMethodload_sql_examples

加载SQL示例

code/C4/03_text2sql_demo_v2.py:101

↓ 1 callersMethodload_table_schemas

加载表结构信息

code/C4/03_text2sql_demo_v2.py:123

↓ 1 callersFunctionmain

()

code/C6/01_llamaindex_evaluation_example.py:19

↓ 1 callersFunctionmain

主函数

code/C4/03_text2sql_demo.py:200

↓ 1 callersFunctionmain

主函数

code/C8/main.py:355

↓ 1 callersFunctionmain

主函数

code/C9/main.py:417

↓ 1 callersFunctionmain

主函数

code/C9/agent(代码系ai生成)/batch_manager.py:198

↓ 1 callersFunctionmain

主函数

code/C9/agent(代码系ai生成)/recipe_ai_agent.py:1298

↓ 1 callersFunctionmain

主函数

code/C9/agent(代码系ai生成)/run_ai_agent.py:102

↓ 1 callersFunctionmain

(argv: list[str])

Extra-chapter/PowerRAG-SDK-Text-QA/code/main.py:40

↓ 1 callersMethodmask

(self, input_ids, vocab_size, device, targets=None, masked_indices=None, probability_matrix=None)

code/C3/visual_bge/visual_bge/eva_clip/hf_model.py:152

↓ 1 callersMethodprocess_recipe

处理单个菜谱

code/C9/agent(代码系ai生成)/recipe_ai_agent.py:596

↓ 1 callersMethodquery_rewrite

智能查询重写 - 让大模型判断是否需要重写查询 Args: query: 原始查询 Returns: 重写后的查询或原查询

code/C8/rag_modules/generation_integration.py:141

↓ 1 callersMethodquery_router

查询路由 - 根据查询类型选择不同的处理方式 Args: query: 用户查询 Returns: 路由类型 ('list', 'detail', 'general')

code/C8/rag_modules/generation_integration.py:205

↓ 1 callersFunctionquery_safe_recursive

(query_str)

code/C3/07_recursive_retrieval_v2.py:64

↓ 1 callersFunctionresize_clip_pos_embed

(state_dict, model, interpolation: str = 'bicubic', seq_dim=1)

code/C3/visual_bge/visual_bge/eva_clip/utils.py:13

↓ 1 callersFunctionresize_eva_pos_embed

(state_dict, model, interpolation: str = 'bicubic', seq_dim=1)

code/C3/visual_bge/visual_bge/eva_clip/utils.py:109

↓ 1 callersFunctionresize_evaclip_pos_embed

(state_dict, model, interpolation: str = 'bicubic', seq_dim=1)

code/C3/visual_bge/visual_bge/eva_clip/utils.py:78

↓ 1 callersFunctionresize_visual_pos_embed

(state_dict, model, interpolation: str = 'bicubic', seq_dim=1)

code/C3/visual_bge/visual_bge/eva_clip/utils.py:46

↓ 1 callersMethodroute_query

智能路由查询到最适合的检索引擎

code/C9/rag_modules/intelligent_query_router.py:167

↓ 1 callersFunctionrun_demo_queries

运行演示查询

code/C4/03_text2sql_demo.py:133

↓ 1 callersMethodrun_interactive

运行交互式问答

code/C8/main.py:310

↓ 1 callersMethodrun_interactive

运行交互式问答

code/C9/main.py:304

↓ 1 callersMethodsave_index

保存向量索引到配置的路径

code/C8/rag_modules/index_construction.py:82

↓ 1 callersMethodsearch

搜索相关内容

code/C4/text2sql/knowledge_base.py:154

↓ 1 callersFunctionsetup_demo

设置演示环境

code/C4/03_text2sql_demo.py:11

↓ 1 callersMethodsetup_llm

初始化大语言模型

code/C8/rag_modules/generation_integration.py:35

↓ 1 callersMethodsetup_retrievers

设置向量检索器和BM25检索器

code/C8/rag_modules/retrieval_optimization.py:30

← previous next →201–300 of 495, ranked by callers