Updating org search API calls

tools/org_seach.py

@@ -7,7 +7,7 @@ from langchain.output_parsers.openai_tools import JsonOutputToolsParser
 from langchain_openai.chat_models import ChatOpenAI
 from langchain_core.runnables import RunnableSequence
 from langchain_core.prompts import ChatPromptTemplate
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 
 try:
     from common.org_search_component import OrgSearch
@@ -18,7 +18,8 @@ search = OrgSearch()
 
 
 class OrganizationNames(BaseModel):
-    orgnames: List[str]
+    """List of names of social-sector organizations, such as nonprofits and foundations."""
+    orgnames: List[str] = Field(description="List of organization names")
 
 
 def extract_org_links_from_chatbot(chatbot_output: str):
@@ -34,7 +35,10 @@ def extract_org_links_from_chatbot(chatbot_output: str):
     Raises:
         ValueError: If parsing fails or if an unexpected output format is received.
     """
-    prompt = """Extract only the names of officially recognized organizations, foundations, and government entities from the text below. Do not include any entries that contain descriptions, regional identifiers, or explanations within parentheses or following the name. Strictly exclude databases, resources, crowdfunding platforms, and general terms. Provide the output only in the specified JSON format.
+    prompt = """Extract only the names of officially recognized organizations, foundations, and government entities
+    from the text below. Do not include any entries that contain descriptions, regional identifiers, or explanations
+    within parentheses or following the name. Strictly exclude databases, resources, crowdfunding platforms, and general
+    terms. Provide the output only in the specified JSON format.
 
     input text below:
     
@@ -121,7 +125,7 @@ def generate_org_link_dict(org_names_list: list):
     for org in org_names_list:
         try:
             # Attempt to retrieve organization data
-            response = search(org)
+            response = search(org, name_only=True)
 
             # Check if there is a valid response and if names are similar
             if response and is_similar(org, response[0].get("names", "")):
@@ -171,7 +175,11 @@ def embed_org_links_in_text(input_text: str, org_link_dict: dict):
                 )
 
         # Append Candid information message at the end
-        input_text += "<p class='candid-app-link'> Visit <a href=https://app.candid.org/ target='_blank' rel='noreferrer' class='candid-org-link'>Candid</a> to get nonprofit information you need.</p>"
+        input_text += (
+            "<p class='candid-app-link'> "
+            "Visit <a href=https://app.candid.org/ target='_blank' rel='noreferrer' class='candid-org-link'>Candid</a> "
+            "to get nonprofit information you need.</p>"
+        )
 
     except TypeError as e:
         print(f"TypeError encountered: {e}")