Functions — github.com/ZBayes/basic_rag

CodeHub MCPcopy Create free account

hub / github.com/ZBayes/basic_rag / functions

Functions46 in github.com/ZBayes/basic_rag

⨍ Functions 46 ◇ Types & classes 11

↓ 7 callersMethodpredict_vec

src/models/vec_model/vec_model.py:23

↓ 3 callersMethodpredict

src/models/llm/llm_model.py:35

↓ 3 callersFunctionrun_client

src/server/client.py:11

↓ 2 callersMethodpost

src/server/handlers/llm_handler.py:31

↓ 2 callersMethodsearch

(self, query, nums=3)

src/searcher/searcher.py:32

↓ 1 callersFunctionStartLlmHandler

(request_config: dict, llm_model: LlmModel)

src/server/handlers/llm_handler.py:39

↓ 1 callersMethod_chat

(self, query: str, history: List[Tuple[str, str]] = None, max_length: int = 2048, num_beams=1, do_

src/models/llm/llm_model.py:30

↓ 1 callersMethod_read_config_

src/models/llm/llm_model.py:21

↓ 1 callersMethodbuild

(self, index_dim)

src/searcher/vec_searcher/vec_index.py:15

↓ 1 callersMethodbuild_llm_prompt

(self, query, retrieval_answer)

src/dm/dialogue_manager.py:43

↓ 1 callersMethodinsert

src/searcher/vec_searcher/vec_index.py:20

↓ 1 callersFunctionlaunch_service

(config, model_mode)

src/server/main_service_online.py:27

↓ 1 callersMethodload

(self, read_path)

src/searcher/vec_searcher/vec_index.py:26

↓ 1 callersMethodpredict_sim

src/models/vec_model/vec_model.py:37

↓ 1 callersMethodpredict_vec_request

src/models/vec_model/vec_model.py:33

↓ 1 callersMethodrank

(self, query, recall_result)

src/searcher/searcher.py:22

↓ 1 callersMethodsave

(self, save_path)

src/searcher/vec_searcher/vec_index.py:30

FunctionStartDialogueManagerHandler

(request_config: dict, dialogue_manager: DialogueManager)

src/server/handlers/dialogue_manager_handler.py:36

FunctionStartSearcherHandler

(request_config: dict, searcher: Searcher)

src/server/handlers/search_handler.py:48

FunctionStartVecModelHandler

(request_config: dict, vec_model: VectorizeModel)

src/server/handlers/vec_model_handler.py:40

src/dm/dialogue_manager.py:19

(self, model_path, config = {})

src/models/llm/llm_model.py:11

(self, pretrained_bert_path, pooling="cls")

src/models/vec_model/simcse_model.py:9

(self, ptm_model_path, device = "cpu")

src/models/vec_model/vec_model.py:11

(self, model_path, vec_search_path)

src/searcher/searcher.py:14

src/searcher/vec_searcher/vec_index.py:12

src/searcher/vec_searcher/vec_searcher.py:6

Methodbatch_insert

src/searcher/vec_searcher/vec_index.py:23

Functionbatch_test

(query_list, process_num, request_num)

src/server/client.py:34

(self, index_dim, index_name)

src/searcher/vec_searcher/vec_searcher.py:11

Functioncal_time_result

src/server/client.py:15

(self, input_ids, attention_mask, token_type_ids)

src/models/vec_model/simcse_model.py:21

Methodinitialize

(self, searcher:Searcher)

src/server/handlers/search_handler.py:20

Methodinitialize

(self, vec_model:VectorizeModel)

src/server/handlers/vec_model_handler.py:20

Methodinitialize

(self, dialogue_manager:DialogueManager)

src/server/handlers/dialogue_manager_handler.py:19

Methodinitialize

(self, llm_model:LlmModel)

src/server/handlers/llm_handler.py:20

(self, vec, doc)

src/searcher/vec_searcher/vec_searcher.py:22

(self, index_name)

src/searcher/vec_searcher/vec_searcher.py:35

src/server/handlers/search_handler.py:35

src/server/handlers/vec_model_handler.py:35

src/server/handlers/dialogue_manager_handler.py:30

src/dm/dialogue_manager.py:23

src/searcher/vec_searcher/vec_searcher.py:28

(self, vec, num)

src/searcher/vec_searcher/vec_index.py:34

(self, vecs, nums = 5)

src/searcher/vec_searcher/vec_searcher.py:47

Functionsingle_test

(url, query_list, num, process_id = 0)

src/server/client.py:21