Skip to content

Commit f7a8ece

Browse files
author
Kavyansh Chourasia
committed
Durable Functions: Refactor AgentPool: Update identifier check for RAG agents to use 'graph-index' for clarity
1 parent aedaa36 commit f7a8ece

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

shiksha-api/durable-functions/core/agents/agent_pool.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def get_rag_agent(cls, identifier: str = "default") -> BaseAzureBlobRAGAgent:
5252
"""
5353
if identifier not in cls._rag_instances:
5454
# TODO: Improve this logic to handle different types of RAG agents
55-
if "graph" in identifier:
55+
if "graph-index" in identifier:
5656
cls._rag_instances[identifier] = cls.GRAPH_INDEX_AGENT_CLASS(identifier)
5757
else:
5858
cls._rag_instances[identifier] = cls.VECTOR_INDEX_AGENT_CLASS(

0 commit comments

Comments
 (0)