增加代码知识库;修复文档处理内容;增加API设置

This commit is contained in:
2026-05-16 20:20:10 +08:00
parent 69b49d28b2
commit 7aa3ce3294
119 changed files with 182273 additions and 793 deletions

View File

@@ -0,0 +1,24 @@
from __future__ import annotations
from typing import List
from app.services.code_kb.adapter import CodeKnowledgeBaseAdapter
from app.services.code_kb.schema import CodeSearchHit
class CodeFunctionRetriever:
def __init__(self, adapter: CodeKnowledgeBaseAdapter) -> None:
self.adapter = adapter
def retrieve(
self,
query: str,
top_k: int = 8,
min_similarity: float = 0.0,
) -> List[CodeSearchHit]:
return self.adapter.search_functions(
query=query,
top_k=top_k,
min_similarity=min_similarity,
)