Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Spinachead/arg
/ functions
Functions
328 in github.com/Spinachead/arg
⨍
Functions
328
◇
Types & classes
68
↳
Endpoints
2
↓ 110 callers
Method
get
(self, key: str)
app/knowledge_base/kb_cache/base.py:74
↓ 16 callers
Function
build_logger
build a logger with colorized output and a log file, for example: logger = build_logger("api") logger.info("<green>some message</green>"
app/utils.py:214
↓ 12 callers
Method
acquire
(self, key: Union[str, Tuple], owner: str = "", msg: str = "")
app/knowledge_base/kb_cache/base.py:90
↓ 12 callers
Method
get_service_by_name
(kb_name: str)
app/knowledge_base/kb_service/base.py:450
↓ 11 callers
Function
get_default_embedding
获取默认嵌入模型
app/utils.py:242
↓ 10 callers
Method
keys
(self)
app/knowledge_base/kb_cache/base.py:66
↓ 10 callers
Method
set
(self, key: str, obj: ThreadSafeObject)
app/knowledge_base/kb_cache/base.py:79
↓ 9 callers
Method
save_vector_store
保存向量库:FAISS保存到磁盘,milvus保存到数据库。PGVector暂未支持
app/knowledge_base/kb_service/base.py:48
↓ 7 callers
Method
exists
(self, kb_name: str = None)
app/knowledge_base/kb_service/base.py:308
↓ 7 callers
Method
load_vector_store
(self)
app/knowledge_base/kb_service/faiss_kb_service.py:29
↓ 7 callers
Method
pop
(self, key: str = None)
app/knowledge_base/kb_cache/base.py:84
↓ 6 callers
Method
load
Load data into document objects.
app/file_rag/document_loaders/FilteredCSVloader.py:33
↓ 5 callers
Method
delete_doc
从知识库删除文件
app/knowledge_base/kb_service/base.py:127
↓ 5 callers
Method
do_add_doc
向知识库添加文档子类实自己逻辑
app/knowledge_base/kb_service/base.py:340
↓ 5 callers
Function
get_Embeddings
(embed_model: str = None)
app/utils.py:96
↓ 5 callers
Function
get_kb_path
(knowledge_base_name: str)
app/knowledge_base/kb_utils.py:54
↓ 5 callers
Method
get_service
( kb_name: str, vector_store_type: Union[str, SupportedVSType], embed_model: str = get
app/knowledge_base/kb_service/base.py:367
↓ 5 callers
Function
handleChange
(id, val)
app/public/elements/KBConfig.jsx:35
↓ 5 callers
Method
list_files
(self)
app/knowledge_base/kb_service/base.py:166
↓ 5 callers
Function
list_files_from_folder
(kb_name: str)
app/knowledge_base/kb_utils.py:80
↓ 5 callers
Function
settings_property
(settings: _T)
app/pydantic_settings_file.py:252
↓ 4 callers
Method
check_embed_model
(self)
app/knowledge_base/kb_service/base.py:54
↓ 4 callers
Method
clear
(self)
app/knowledge_base/kb_cache/faiss_cache.py:43
↓ 4 callers
Method
create_kb
创建知识库
app/knowledge_base/kb_service/base.py:57
↓ 4 callers
Function
file_to_kbfile
(kb_name: str, files: List[str])
app/knowledge_base/migrate.py:99
↓ 4 callers
Function
get_file_path
(knowledge_base_name: str, doc_name: str)
app/knowledge_base/kb_utils.py:65
↓ 4 callers
Function
list_kbs_from_db
(session, min_file_count: int = -1)
app/db/repository/knowledge_base_repository.py:28
↓ 4 callers
Function
session_scope
上下文管理器用于自动获取 Session, 避免错误
app/db/session.py:18
↓ 4 callers
Method
vs_type
(self)
app/knowledge_base/kb_service/base.py:313
↓ 3 callers
Method
add_doc
向知识库添加文件 如果指定了docs,则不再将文本向量化,并将数据库对应条目标为custom_docs=True
app/knowledge_base/kb_service/base.py:88
↓ 3 callers
Method
clear_vs
删除向量库中所有内容
app/knowledge_base/kb_service/base.py:72
↓ 3 callers
Function
files2docs_in_thread
利用多线程批量将磁盘文件转化成langchain Document. 如果传入参数是Tuple,形式为(filename, kb_name) 生成器返回值为 status, (kb_name, file_name, docs | error)
app/knowledge_base/kb_utils.py:458
↓ 3 callers
Function
files2vs
(kb_name: str, kb_files: List[KnowledgeFile])
app/knowledge_base/migrate.py:129
↓ 3 callers
Function
get_doc_path
(knowledge_base_name: str)
app/knowledge_base/kb_utils.py:57
↓ 3 callers
Function
get_file_detail
(session, kb_name: str, filename: str)
app/db/repository/knowledge_file_repository.py:220
↓ 3 callers
Function
get_ocr
(use_cuda: bool = True)
app/file_rag/document_loaders/ocr.py:13
↓ 3 callers
Function
get_vs_path
(knowledge_base_name: str, vector_name: str)
app/knowledge_base/kb_utils.py:61
↓ 3 callers
Method
to_msg_template
(self, is_raw=True)
app/utils.py:176
↓ 2 callers
Method
__read_file
(self, csvfile: TextIOWrapper)
app/file_rag/document_loaders/FilteredCSVloader.py:59
↓ 2 callers
Function
_clean_document_content
清洗文档内容,移除控制字符和特殊 Unicode
app/logic/action.py:89
↓ 2 callers
Function
add_file_to_db
( session, kb_file: KnowledgeFile, docs_count: int = 0, custom_docs: bool = False, doc_inf
app/db/repository/knowledge_file_repository.py:111
↓ 2 callers
Function
add_kb_to_db
(session, kb_name, kb_info, vs_type, embed_model)
app/db/repository/knowledge_base_repository.py:7
↓ 2 callers
Method
create_yaml_template
generate yaml template with default object sub_comments indicate how to populate comments for sub models, it could be nested.
app/pydantic_settings_file.py:103
↓ 2 callers
Method
embed_documents
Embed search docs. Args: texts: List of text to embed. Returns: List of embeddings.
app/utils.py:52
↓ 2 callers
Method
file2docs
(self, refresh: bool = False)
app/knowledge_base/kb_utils.py:357
↓ 2 callers
Method
file2text
( self, zh_title_enhance: bool = False, refresh: bool = False,
app/knowledge_base/kb_utils.py:412
↓ 2 callers
Method
finish_loading
(self)
app/knowledge_base/kb_cache/base.py:45
↓ 2 callers
Method
from_vectorstore
( vectorstore: VectorStore, top_k: int, score_threshold: int | float, )
app/file_rag/retrievers/base.py:18
↓ 2 callers
Function
get_Retriever
(type: str = "vectorstore")
app/file_rag/fr_utils.py:15
↓ 2 callers
Method
get_class_comment
you can override this to customize class comments
app/pydantic_settings_file.py:80
↓ 2 callers
Method
get_field_comment
you can override this to customize field comments model_obj is the instance that field_name belongs to
app/pydantic_settings_file.py:88
↓ 2 callers
Method
get_relevant_documents
(self, query: str)
app/file_rag/retrievers/base.py:26
↓ 2 callers
Function
get_user_by_identifier
根据用户标识符获取用户信息
app/db/repository/user_repository.py:9
↓ 2 callers
Function
import_yaml
()
app/pydantic_settings_file.py:21
↓ 2 callers
Function
list_docs_from_db
列出某知识库某文件对应的所有Document。 返回形式:[{"id": str, "metadata": dict}, ...]
app/db/repository/knowledge_file_repository.py:28
↓ 2 callers
Function
list_kbs_from_folder
()
app/knowledge_base/kb_utils.py:72
↓ 2 callers
Function
run_in_thread_pool
在线程池中批量运行任务,并将运行结果以生成器的形式返回。 请确保任务中的所有操作是线程安全的,任务函数请全部使用关键字参数。
app/utils.py:267
↓ 2 callers
Method
save
(self, path: str, create_path: bool = True)
app/knowledge_base/kb_cache/faiss_cache.py:35
↓ 2 callers
Method
update_doc
使用content中的文件更新向量库 如果指定了docs,则使用自定义docs,并将数据库对应条目标为custom_docs=True
app/knowledge_base/kb_service/base.py:149
↓ 2 callers
Function
upload_document
(kb_name: str)
app/logic/action.py:19
↓ 1 callers
Method
__init__
( self, model_obj: BaseModel, dump_kwds: t.Dict={}, sub_comments: t.Dict[str,
app/pydantic_settings_file.py:49
↓ 1 callers
Method
__init__
对应知识库目录中的文件,必须是磁盘上存在的才能进行向量化等操作。
app/knowledge_base/kb_utils.py:335
↓ 1 callers
Function
_cached_settings
the sesstings is cached, and refreshed when configuration files changed
app/pydantic_settings_file.py:243
↓ 1 callers
Method
_check_count
(self)
app/knowledge_base/kb_cache/base.py:69
↓ 1 callers
Method
_create_yaml_object
helper method to convert settings instance to ruamel.YAML object
app/pydantic_settings_file.py:63
↓ 1 callers
Function
_get_result_to_documents
(get_result: GetResult)
app/knowledge_base/kb_service/chromadb_kb_service.py:19
↓ 1 callers
Method
_load_chroma
(self)
app/knowledge_base/kb_service/chromadb_kb_service.py:69
↓ 1 callers
Function
_save_files_in_thread
通过多线程将上传的文件保存到对应知识库目录内。 生成器返回保存结果:{"code":200, "msg": "xxx", "data": {"knowledge_base_name":"xxx", "file_name": "xxx"}}
app/knowledge_base/kb_doc_api.py:111
↓ 1 callers
Function
_split_text_with_regex_from_end
( text: str, separator: str, keep_separator: bool )
app/file_rag/text_splitter/chinese_recursive_text_splitter.py:8
↓ 1 callers
Function
add_docs_to_db
将某知识库某文件对应的所有Document信息添加到数据库。 doc_infos形式:[{"id": str, "metadata": dict}, ...]
app/db/repository/knowledge_file_repository.py:67
↓ 1 callers
Function
add_message_to_db
保存对话消息到数据库
app/db/repository/message_repository.py:7
↓ 1 callers
Function
add_user_memory
添加用户记忆 :param session: 数据库会话 :param user_id: 用户ID :param memory_text: 记忆内容摘要 :param importance: 重要程度 1-5 :param metadata: 元数据
app/db/repository/user_memory_repository.py:12
↓ 1 callers
Function
build_knowledge_graph
()
app/core/state_graph/knowledge_query_graph.py:9
↓ 1 callers
Function
build_main_graph
()
app/core/state_graph/main_graph.py:49
↓ 1 callers
Function
build_sql_query_graph
()
app/core/state_graph/sql_query_graph.py:7
↓ 1 callers
Function
count_files_from_db
(session, kb_name: str)
app/db/repository/knowledge_file_repository.py:91
↓ 1 callers
Function
create_kb
( knowledge_base_name: str = None, vector_store_type: str = None, kb_info: str = "",
app/knowledge_base/kb_api.py:19
↓ 1 callers
Function
create_memory_from_conversation
从对话历史创建记忆(供后台任务或定期汇总使用) :param session: 数据库会话 :param user_id: 用户ID :param conversation_summary: 对话摘要 :param kb_name: 使用的知识库名称
app/db/repository/user_memory_repository.py:215
↓ 1 callers
Function
create_tables
()
app/knowledge_base/migrate.py:46
↓ 1 callers
Method
del_doc_by_ids
(self, ids: List[str])
app/knowledge_base/kb_service/base.py:186
↓ 1 callers
Function
delete_docs_from_db
删除某知识库某文件对应的所有Document,并返回被删除的Document。 返回形式:[{"id": str, "metadata": dict}, ...]
app/db/repository/knowledge_file_repository.py:48
↓ 1 callers
Function
delete_file_from_db
(session, kb_file: KnowledgeFile)
app/db/repository/knowledge_file_repository.py:161
↓ 1 callers
Function
delete_files_from_db
(session, knowledge_base_name: str)
app/db/repository/knowledge_file_repository.py:187
↓ 1 callers
Function
delete_kb_from_db
(session, kb_name)
app/db/repository/knowledge_base_repository.py:64
↓ 1 callers
Method
do_clear_vs
从知识库删除全部向量子类实自己逻辑
app/knowledge_base/kb_service/base.py:358
↓ 1 callers
Method
do_create_kb
创建知识库子类实自己逻辑
app/knowledge_base/kb_service/base.py:247
↓ 1 callers
Method
do_delete_doc
从知识库删除文档子类实自己逻辑
app/knowledge_base/kb_service/base.py:351
↓ 1 callers
Method
do_drop_kb
删除知识库子类实自己逻辑
app/knowledge_base/kb_service/base.py:321
↓ 1 callers
Method
do_drop_kb
(self)
app/knowledge_base/kb_service/faiss_kb_service.py:57
↓ 1 callers
Method
do_init
(self, **kwargs)
app/file_rag/retrievers/base.py:14
↓ 1 callers
Method
do_init
(self)
app/knowledge_base/kb_service/base.py:317
↓ 1 callers
Method
do_init
(self)
app/knowledge_base/kb_service/default_kb_service.py:40
↓ 1 callers
Method
do_search
搜索知识库子类实自己逻辑
app/knowledge_base/kb_service/base.py:328
↓ 1 callers
Method
docs2texts
( self, docs: List[Document] = None, zh_title_enhance: bool = False,
app/knowledge_base/kb_utils.py:373
↓ 1 callers
Method
docs_count
(self)
app/knowledge_base/kb_cache/faiss_cache.py:32
↓ 1 callers
Method
drop_kb
删除知识库
app/knowledge_base/kb_service/base.py:80
↓ 1 callers
Method
exist_doc
(self, file_name: str)
app/knowledge_base/kb_service/base.py:161
↓ 1 callers
Function
file_exists_in_db
(session, kb_file: KnowledgeFile)
app/db/repository/knowledge_file_repository.py:207
↓ 1 callers
Function
format_reference
将知识库检索结果格式化为参考文档的格式
app/utils.py:123
↓ 1 callers
Function
generate_queries_step
(data: dict)
app/logic/onMessage.py:14
next →
1–100 of 328, ranked by callers