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
Function
delete_user_memory
删除用户记忆 :param session: 数据库会话 :param memory_id: 记忆ID :return: 是否删除成功
app/db/repository/user_memory_repository.py:132
Method
do_add_doc
(self, docs: List[Document], **kwargs)
app/knowledge_base/kb_service/chromadb_kb_service.py:110
Method
do_add_doc
( self, docs: List[Document], **kwargs, )
app/knowledge_base/kb_service/faiss_kb_service.py:84
Method
do_add_doc
(self, docs: List[Document], **kwargs)
app/knowledge_base/kb_service/default_kb_service.py:31
Method
do_clear_vs
(self)
app/knowledge_base/kb_service/chromadb_kb_service.py:148
Method
do_clear_vs
(self)
app/knowledge_base/kb_service/faiss_kb_service.py:120
Method
do_clear_vs
(self)
app/knowledge_base/kb_service/default_kb_service.py:34
Method
do_create_kb
(self)
app/knowledge_base/kb_service/chromadb_kb_service.py:83
Method
do_create_kb
(self)
app/knowledge_base/kb_service/faiss_kb_service.py:52
Method
do_create_kb
(self)
app/knowledge_base/kb_service/default_kb_service.py:25
Method
do_delete_doc
(self, kb_file: KnowledgeFile, **kwargs)
app/knowledge_base/kb_service/chromadb_kb_service.py:157
Method
do_delete_doc
(self, kb_file: KnowledgeFile, **kwargs)
app/knowledge_base/kb_service/faiss_kb_service.py:107
Method
do_delete_doc
(self, kb_file, **kwargs)
app/knowledge_base/kb_service/default_kb_service.py:52
Method
do_drop_kb
(self)
app/knowledge_base/kb_service/chromadb_kb_service.py:86
Method
do_drop_kb
(self)
app/knowledge_base/kb_service/default_kb_service.py:28
Method
do_init
( self, retriever: BaseRetriever = None, top_k: int = 5, )
app/file_rag/retrievers/vectorstore.py:14
Method
do_init
( self, retriever: BaseRetriever = None, top_k: int = 5, cache_key: str = "def
app/file_rag/retrievers/ensemble.py:86
Method
do_init
( self, retriever: BaseRetriever = None, top_k: int = 5, )
app/file_rag/retrievers/milvus_vectorstore.py:98
Method
do_init
(self)
app/knowledge_base/kb_service/chromadb_kb_service.py:76
Method
do_init
(self)
app/knowledge_base/kb_service/faiss_kb_service.py:47
Method
do_insert_multi_knowledge
(self)
app/knowledge_base/kb_service/default_kb_service.py:46
Method
do_insert_one_knowledge
(self)
app/knowledge_base/kb_service/default_kb_service.py:49
Method
do_search
( self, query: str, top_k: int, score_threshold: float = Settings.kb_settings.SCORE_THRESHOLD )
app/knowledge_base/kb_service/chromadb_kb_service.py:97
Method
do_search
( self, query: str, top_k: int, score_threshold: float = 2.0, )
app/knowledge_base/kb_service/faiss_kb_service.py:64
Method
do_search
(self, query, top_k, score_threshold)
app/knowledge_base/kb_service/default_kb_service.py:43
Method
doc2text
(filepath)
app/file_rag/document_loaders/mydocloader.py:8
Method
embed_query
Embed query text. Args: text: Text to embed. Returns: Embedding.
app/utils.py:63
Function
execute
()
app/logic/onChatStart.py:16
Function
execute
(message: cl.Message)
app/logic/onMessage.py:32
Function
execute
(username: str, password: str)
app/logic/authCallback.py:5
Function
execute1
(message: cl.Message)
app/logic/onMessage.py:112
Function
execute_sql
执行SQL查询
app/core/state_graph/nodes/sql_query_graph/execute_sql.py:4
Function
execute_sql_query
执行sql查询语句并且返回结果. Args: sql: The SQL query string to execute
app/core/state_graph/tool/tools.py:38
Function
execute_tools
执行工具调用并收集结果
app/weather.py:47
Method
exist_doc
(self, file_name: str)
app/knowledge_base/kb_service/faiss_kb_service.py:129
Method
file_exist
(self)
app/knowledge_base/kb_utils.py:432
Function
files2docs_in_thread_file2docs
( *, file: KnowledgeFile, **kwargs )
app/knowledge_base/kb_utils.py:442
Function
folder2db
use existed files in local folder to populate database and/or vector store. set parameter `mode` to: recreate_vs: recreate all vector
app/knowledge_base/migrate.py:111
Method
from_data
(cls, h: Union[List, Tuple, Dict])
app/utils.py:194
Method
from_vectorstore
( vectorstore: VectorStore, top_k: int, score_threshold: int | float, )
app/file_rag/retrievers/vectorstore.py:24
Method
from_vectorstore
( vectorstore: VectorStore, top_k: int, score_threshold: int | float, )
app/file_rag/retrievers/ensemble.py:98
Method
from_vectorstore
( vectorstore: VectorStore, top_k: int, score_threshold: int or float, )
app/file_rag/retrievers/milvus_vectorstore.py:108
Function
generate_answer
把sql查询结果生成自然语言回答
app/core/state_graph/nodes/sql_query_graph/generate_answer.py:10
Function
generate_queries
生成多个查询变体以及其对应的知识库名称
app/core/state_graph/nodes/knowledge_query_graph/generate_queries.py:12
Function
generate_sql
生成sql查询语句，并调用工具执行
app/core/state_graph/nodes/sql_query_graph/generate_sql.py:9
Function
get_current_weather
Get current weather for a city.
app/weather.py:18
Function
get_data_layer
()
app/app.py:38
Function
get_db
()
app/db/session.py:46
Function
get_db0
()
app/db/session.py:54
Method
get_default
()
app/knowledge_base/kb_service/base.py:457
Method
get_doc_by_ids
(self, ids: List[str])
app/knowledge_base/kb_service/chromadb_kb_service.py:140
Method
get_doc_by_ids
(self, ids: List[str])
app/knowledge_base/kb_service/faiss_kb_service.py:39
Function
get_kb_detail
(session, kb_name: str)
app/db/repository/knowledge_base_repository.py:76
Function
get_kb_details
()
app/knowledge_base/kb_service/base.py:461
Function
get_message_by_id
根据消息 ID 获取单条消息
app/db/repository/message_repository.py:57
Function
get_messages_by_thread_id
根据 threadId 查询对话历史
app/db/repository/message_repository.py:36
Method
get_relative_source_path
将文件路径转化为相对路径，保证查询时一致
app/knowledge_base/kb_service/base.py:230
Method
get_relevant_documents
(self, query: str)
app/file_rag/retrievers/vectorstore.py:40
Method
get_relevant_documents
(self, query: str)
app/file_rag/retrievers/ensemble.py:190
Method
get_relevant_documents
(self, query: str)
app/file_rag/retrievers/milvus_vectorstore.py:120
Function
get_sn_table_count
Get the total number of records in the 'sn' table.
app/core/state_graph/tool/tools.py:8
Function
get_user_memory_by_id
根据ID获取单条记忆 :param session: 数据库会话 :param memory_id: 记忆ID :return: 记忆对象或None
app/db/repository/user_memory_repository.py:71
Function
get_user_profile
()
app/knowledge_base/kb_doc_api.py:295
Method
img2text
(filepath)
app/file_rag/document_loaders/myimgloader.py:10
Function
import_from_db
在知识库与向量库无变化的情况下，从备份数据库中导入数据到 info.db。 适用于版本升级时，info.db 结构变化，但无需重新向量化的情况。 请确保两边数据库表名一致，需要导入的字段名一致 当前仅支持 sqlite
app/knowledge_base/migrate.py:55
Function
init_default_kb
()
app/knowledge_base/kb_api.py:102
Method
invalidate
使指定缓存失效
app/file_rag/retrievers/ensemble.py:69
Method
iter_block_items
(parent)
app/file_rag/document_loaders/mydocloader.py:32
Method
key
(self)
app/knowledge_base/kb_cache/base.py:26
Function
list_file_num_docs_id_by_kb_name_and_file_name
列出某知识库某文件对应的所有Document的id。 返回形式：[str, ...]
app/db/repository/knowledge_file_repository.py:10
Function
list_files
( knowledge_base_name:str )
app/knowledge_base/kb_api.py:80
Function
list_kbs
()
app/knowledge_base/kb_api.py:16
Method
list_kbs_type
()
app/knowledge_base/kb_service/base.py:254
Function
list_user_memories
获取用户记忆列表，按重要程度和最近使用时间排序 :param session: 数据库会话 :param user_id: 用户ID :param limit: 返回数量限制 :return: 记忆列表
app/db/repository/user_memory_repository.py:45
Method
load_vector_store
( self, kb_name: str, vector_name: str = None, create: bool = True, em
app/knowledge_base/kb_cache/faiss_cache.py:90
Method
load_vector_store
( self, kb_name: str, embed_model: str = get_default_embedding(), )
app/knowledge_base/kb_cache/faiss_cache.py:147
Function
main
(message: cl.Message)
app/app.py:30
Function
mcp_tool_node
处理 MCP 工具调用
app/core/state_graph/tool/mcp_tool_node.py:9
Method
model_cls
(self)
app/pydantic_settings_file.py:60
Method
model_post_init
(self, __context: os.Any)
app/pydantic_settings_file.py:183
Method
obj
(self)
app/knowledge_base/kb_cache/base.py:52
Function
on_action
(action: cl.Action)
app/app.py:44
Function
on_logout
Called when a user logs out
app/app.py:109
Function
on_mcp_connect
Called when an MCP connection is established
app/app.py:62
Function
on_mcp_disconnect
Called when an MCP connection is terminated
app/app.py:78
Function
on_resume
(thread: ThreadDict)
app/app.py:48
Function
on_upload_document_action
(action: cl.Action)
app/app.py:52
Method
pdf2text
(filepath)
app/file_rag/document_loaders/mypdfloader.py:45
Function
prune_db_docs
delete docs in database that not existed in local folder. it is used to delete database docs after user deleted some doc files in file browse
app/knowledge_base/migrate.py:215
Function
prune_folder_files
delete doc files in local folder that not existed in database. it is used to free local disk space by delete unused doc files.
app/knowledge_base/migrate.py:233
Function
reset_tables
()
app/knowledge_base/migrate.py:50
Function
respond
调用 LLM 生成回复，支持 MCP 工具调用
app/core/state_graph/nodes/knowledge_query_graph/respond.py:34
Function
respond
根据上下文调用LLM大模型进行回复
app/core/state_graph/nodes/main_graph/respond.py:10
Function
respond_to_general_query
Generates a response to a general user query based on the agent's current state and routing logic.
app/core/state_graph/nodes/main_graph/respond_to_general_query.py:9
Function
retrieve_documents
使用多个查询变体检索文档并合并结果
app/core/state_graph/nodes/knowledge_query_graph/retrieve_documents.py:10
Method
rotate_img
img --image angle --rotation angle return--rotated img
app/file_rag/document_loaders/mypdfloader.py:21
Function
route_query
(state: MainState)
app/weather.py:86
Function
route_query
Determines the next action for the agent based on the router type in the current state. Args: state (AgentState): The current state
app/core/state_graph/nodes/main_graph/route_query.py:5
Function
route_tools
判断是否需要调用工具，并路由到对应的工具节点
app/core/state_graph/nodes/sql_query_graph/route_tools.py:5
Function
route_tools
判断是否需要调用工具，并路由到对应的工具节点
app/core/state_graph/nodes/main_graph/route_tools.py:5
← previous
next →
201–300 of 328, ranked by callers