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

↓ 1 callersFunctionshow_batch_details

显示批次详细信息

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

↓ 1 callersFunctionshow_help

显示帮助信息

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

↓ 1 callersFunctionshow_progress_status

显示处理进度状态

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

↓ 1 callersMethodsimilarity_search

相似度搜索 Args: query: 查询文本 k: 返回结果数量 filters: 过滤条件 Returns:

code/C9/rag_modules/milvus_index_construction.py:325

↓ 1 callersMethodstem

(self, x)

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

↓ 1 callersFunctiontest_single_recipe

测试单个菜谱解析

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

↓ 1 callersMethodtopic_level_retrieval

主题级检索：专注于广泛主题和概念使用图索引的关系键值对结构进行主题检索

code/C9/rag_modules/hybrid_retrieval.py:319

↓ 1 callersMethodunderstand_graph_query

理解查询的图结构意图这是图RAG的核心：从自然语言到图查询的转换

code/C9/rag_modules/graph_rag_retrieval.py:141

↓ 1 callersMethodvector_search_enhanced

增强的向量检索：结合图信息

code/C9/rag_modules/hybrid_retrieval.py:518

↓ 1 callersMethodvectorize_and_store

向量化数据并存储到Milvus

code/C4/03_text2sql_demo_v2.py:175

↓ 1 callersMethodvisualize_comparison

可视化对比不同搜索模式

code/C3/work_hybrid_multimodal_search.py:339

↓ 1 callersFunctionvisualize_results

从检索到的结果创建一个全景图用于可视化

code/C3/work_multimodal_dragon_search.py:94

↓ 1 callersFunctionvisualize_results

从检索到的结果创建一个全景图用于可视化

code/C3/work_hybrid_multimodal_search.py:97

↓ 1 callersFunctionvisualize_results

从检索到的图像列表创建一个全景图用于可视化。

code/C3/04_multi_milvus.py:35

Method__call__

主调用方法，返回密集和稀疏向量

code/C4/01_hybrid_search_v2.py:100

Method__call__

(self, texts:Union[str, List[str]], context_length:int=77)

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

Method__del__

析构函数，确保关闭连接

code/C9/rag_modules/graph_data_preparation.py:456

Method__del__

析构函数

code/C9/rag_modules/milvus_index_construction.py:501

Method__init__

初始化SigLIP嵌入函数 Args: model_name: SigLIP模型名称 device: 设备类型 ("cpu" 或 "cuda")

code/C4/01_hybrid_search_v2.py:18

Method__init__

(self, **kwargs)

code/C4/work_rerank_and_refine.py:22

Method__init__

(self, model_name="BAAI/bge-small-zh-v1.5", device="cpu")

code/C4/03_text2sql_demo_v2.py:15

Method__init__

(self, milvus_uri: str = "http://localhost:19530")

code/C4/03_text2sql_demo_v2.py:44

Method__init__

(self, **kwargs)

code/C4/07_rerank_and_refine.py:22

Method__init__

初始化代理

code/C4/text2sql/text2sql_agent.py:11

Method__init__

(self, api_key: str = None)

code/C4/text2sql/sql_generator.py:10

Method__init__

(self, milvus_uri: str = "http://localhost:19530")

code/C4/text2sql/knowledge_base.py:11

Method__init__

初始化RAG系统 Args: config: RAG系统配置，默认使用DEFAULT_CONFIG

code/C8/main.py:36

Method__init__

初始化检索优化模块 Args: vectorstore: FAISS向量存储 chunks: 文档块列表

code/C8/rag_modules/retrieval_optimization.py:18

Method__init__

初始化生成集成模块 Args: model_name: 模型名称 temperature: 生成温度 max_tokens: 最大token数

code/C8/rag_modules/generation_integration.py:20

Method__init__

初始化索引构建模块 Args: model_name: 嵌入模型名称 index_save_path: 索引保存路径

code/C8/rag_modules/index_construction.py:18

Method__init__

初始化数据准备模块 Args: data_path: 数据文件夹路径

code/C8/rag_modules/data_preparation.py:33

Method__init__

(self, data_dir: str, metadata_path: str)

code/C3/work_multimodal_dragon_search.py:29

Method__init__

(self, model_name: str, model_path: str)

code/C3/work_multimodal_dragon_search.py:71

Method__init__

(self, data_dir: str, metadata_path: str)

code/C3/work_hybrid_multimodal_search.py:29

Method__init__

(self, visual_model_name: str, visual_model_path: str)

code/C3/work_hybrid_multimodal_search.py:60

Method__init__

(self, data_dir: str, metadata_path: str, collection_name: str, milvus_uri: str)

code/C3/work_hybrid_multimodal_search.py:135

Method__init__

(self, model_name: str, model_path: str)

code/C3/04_multi_milvus.py:21

Method__init__

(self, model_name_bge: str = None, model_weight = None, # "/path/to/your/wei

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

Method__init__

( self, local_loss=False, gather_with_grad=False, cache_labels

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

Method__init__

( self, dim, pt_seq_len, ft_seq_len=None, custom_freqs = None,

code/C3/visual_bge/visual_bge/eva_clip/rope.py:80

Method__init__

(self, spacial_dim: int, embed_dim: int, num_heads: int, output_dim: int = None)

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

Method__init__

(self, layers, output_dim, heads, image_size=224, width=64)

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

Method__init__

(self, bpe_path: str = default_bpe(), special_tokens=None)

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

Method__init__

(self, tokenizer_name:str)

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

Method__init__

(self, max_size, interpolation=InterpolationMode.BICUBIC, fn='max', fill=0)

code/C3/visual_bge/visual_bge/eva_clip/transform.py:15

Method__init__

( self, model_name, embed_dim, image_size=224, poo

code/C3/visual_bge/visual_bge/eva_clip/timm_model.py:33

Method__init__

( self, model_name_or_path: str, output_dim: int, tokenizer_n

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

Method__init__

(self, *args, **kwargs)

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

Method__init__

(self, dim, init_values=1e-5, inplace=False)

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

Method__init__

(self, prob, exclude_first_token=True)

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

Method__init__

( self, dim, num_heads=8, qkv_bias=True, scaled_co

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

Method__init__

( self, d_model: int, n_head: int, mlp_ratio: float = 4.0,

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

Method__init__

( self, width: int, layers: int, heads: int, mlp_r

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

Method__init__

( self, d_model: int, n_head: int, mlp_ratio: float = 4.0,

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

Method__init__

( self, width: int, layers: int, heads: int, mlp_r

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

Method__init__

( self, image_size: int, patch_size: int, width: int,

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

Method__init__

( self, context_length: int = 77, vocab_size: int = 49408, wid

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

Method__init__

(self, drop_prob=None)

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

Method__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.SiLU, drop=0., norm

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

Method__init__

( self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., w

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

Method__init__

(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr

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

Method__init__

(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)

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

Method__init__

(self, window_size, num_heads)

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

Method__init__

(self, img_size=224, patch_size=16, in_chans=3, num_classes=1000, embed_dim=768, depth=12, nu

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

Method__init__

( self, embed_dim: int, vision_cfg: CLIPVisionCfg, text_cfg: C

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

Method__init__

(self, config: Optional[GraphRAGConfig] = None)

code/C9/main.py:45

Method__init__

(self, api_key: str, base_url: str = "https://api.moonshot.cn/v1")

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

Method__init__

(self, ai_agent: KimiRecipeAgent, output_dir: str = "./ai_output", batch_size: int = 20)

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

Method__init__

(self)

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

Method__init__

(self, config, milvus_module, data_module, llm_client)

code/C9/rag_modules/hybrid_retrieval.py:53

Method__init__

初始化生成集成模块

code/C9/rag_modules/generation_integration.py:18

Method__init__

(self, traditional_retrieval, # 传统混合检索模块 graph_rag_retrieval, # 图RAG检索模

code/C9/rag_modules/intelligent_query_router.py:47

Method__init__

初始化图数据库连接 Args: uri: Neo4j连接URI user: 用户名 password: 密码 database: 数据库名称

code/C9/rag_modules/graph_data_preparation.py:34

Method__init__

(self, config, llm_client)

code/C9/rag_modules/graph_rag_retrieval.py:66

Method__init__

初始化Milvus索引构建模块 Args: host: Milvus服务器地址 port: Milvus服务器端口 collection_name: 集合名称 dime

code/C9/rag_modules/milvus_index_construction.py:19

Method__init__

(self, config, llm_client)

code/C9/rag_modules/graph_indexing.py:48

Method__post_init__

初始化后的处理

code/C8/config.py:27

Method__post_init__

初始化后的处理

code/C9/config.py:45

Method__post_init__

(self)

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

Function_convert_to_rgb

(image)

code/C3/visual_bge/visual_bge/eva_clip/transform.py:39

Function_convert_weights

(l)

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

Method_fallback_search

降级搜索方法（简单文本匹配）

code/C4/03_text2sql_demo_v2.py:253

Method_init_weights

(self, m)

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

Method_unlock

(x)

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

Methodadaptive_query_planning

自适应查询规划：根据查询复杂度动态调整策略

code/C9/rag_modules/graph_rag_retrieval.py:435

Methodadd_documents

向现有索引添加新文档 Args: new_chunks: 新的文档块列表 Returns: 是否添加成功

code/C9/rag_modules/milvus_index_construction.py:273

Methodadd_example

添加新的Q->SQL示例

code/C4/text2sql/text2sql_agent.py:132

Functionadd_model_config

add model config path or file and update registry

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

Methodadd_sql_example

添加新的SQL示例

code/C4/03_text2sql_demo_v2.py:282

Methodbackward

(ctx, grad_output)

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

Methodcleanup

清理资源

code/C4/text2sql/text2sql_agent.py:205

Methodcleanup

清理资源

code/C4/text2sql/knowledge_base.py:179

Methodclose

关闭资源连接

code/C9/rag_modules/hybrid_retrieval.py:795

Methodcompress_documents

对文档进行ColBERT重排序

code/C4/work_rerank_and_refine.py:78

Methodcompress_documents

对文档进行ColBERT重排序

code/C4/07_rerank_and_refine.py:78

Functioncreate_model_and_transforms

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

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

Functioncreate_model_from_pretrained

( model_name: str, pretrained: str, precision: str = 'fp32', device: Union[str

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

Methoddim

返回向量维度

code/C4/03_text2sql_demo_v2.py:36

Methodencode_query

编码查询，支持多种模式

code/C3/work_hybrid_multimodal_search.py:83

Methodencode_text

ColBERT文本编码

code/C4/work_rerank_and_refine.py:33

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