apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-04-26 08:56:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add text rag source

Browse source
This commit is contained in:
Ramnique Singh 2025-03-27 22:44:28 +05:30
parent e317c1eb50
commit a56b2d3a90
10 changed files with 532 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

7
apps/rowboat_agents/src/graph/tool_calling.py
View file

@ -15,7 +15,10 @@ data_sources_collection = db['sources']
data_source_docs_collection = db['source_docs']
qdrant_client = QdrantClient(url=os.environ.get("QDRANT_URL"))
qdrant_client = QdrantClient(
url=os.environ.get("QDRANT_URL"),
api_key=os.environ.get("QDRANT_API_KEY") or None
)
# Initialize OpenAI client
client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
@ -65,7 +68,7 @@ async def call_rag_tool(
# Create embedding for the query
embed_result = await embed(model=embedding_model, value=query)
print(embed_result)
# print(embed_result)
# Fetch all active data sources for this project
sources = await data_sources_collection.find({
"projectId": project_id,