Update ComorbidityCheckerAgent.py

ComorbidityCheckerAgent.py

@@ -10,6 +10,14 @@ from langchain_openai import ChatOpenAI
 from langchain_community.vectorstores import Chroma
 from embedding_manager import DirectoryEmbeddingManager
 
+SEED_SOURCES = [
+    "https://www.cms.gov/medicare/payment/medicare-advantage-rates-statistics/risk-adjustment",
+    "https://www.cms.gov/data-research/monitoring-programs/medicare-risk-adjustment-data-validation-program",
+    "https://www.cms.gov/files/document/fy-2024-icd-10-cm-coding-guidelines-updated-02/01/2024.pdf",
+    "https://www.aapc.com/blog/41212-include-meat-in-your-risk-adjustment-documentation/",
+]
+
+
 class ComorbidityCheckerAgent:
     """
     Uses a two-step AI agent process to first identify potential comorbidities for a
@@ -55,7 +63,7 @@ class ComorbidityCheckerAgent:
                 "Your primary skill is to research disease patterns and then meticulously cross-reference them with patient records embeddings "
                 "to ensure accurate documentation and coding. You provide clear, evidence-based findings."
             ),
-            tools=[patient_chart_search],
+            tools=[self.search_tool,patient_chart_search],
             verbose=True,
             memory=False,
             llm=self.llm,