Update app.py

app.py

@@ -9,7 +9,7 @@ from xml.etree import ElementTree
 
 @tool
 def get_lastest_filing_info(company: str = None, cik: str = None, type: str = "10-K", dateb: str = None, start: str = '0', count: str = '50') -> str:
-    """A tool that gets the most recent 10-K filings of a company from the SEC.
+    """A tool that gets the most recent 10-K filings (including addendums) of a company from the SEC.
     Args:
         company: The name of the company.
         cik: The Central Index Key (CIK) of the company.
@@ -24,8 +24,9 @@ def get_lastest_filing_info(company: str = None, cik: str = None, type: str = "1
     if dateb:
         try:
             datetime.datetime.strptime(dateb, "%Y%m%d")
-        except ValueError:
-            return {"error": "Invalid dateb format. Use YYYYMMDD."}
+        except ValueError as e:
+            print(f"Date validation error: {e}")
+            return {"error": str(e)}
     
     # Ensure dateb is not after the current time
     current_date = datetime.datetime.now().strftime("%Y%m%d")
@@ -43,10 +44,15 @@ def get_lastest_filing_info(company: str = None, cik: str = None, type: str = "1
         "count": count,
         "output": "atom"
     }
+    # Set a proper User-Agent header per SEC guidelines
+    headers = {
+        "User-Agent": "FuncPhenomenon SMOLAgentTest/1.0"
+    }
     
-    response = requests.get(base_url, params=params)
+    response = requests.get(base_url, params=params, headers=headers)
     if response.status_code != 200:
-        return {"error": "Failed to retrieve data from SEC"}
+        print(f"SEC retrieval error: {response.text}")
+        return {"error": response.text}
     
     filings = []
     try:
@@ -69,8 +75,9 @@ def get_lastest_filing_info(company: str = None, cik: str = None, type: str = "1
         # Fetch additional pages until we have 100 filings or no more are available
         while len(filings) < 100:
             params["start"] = str(int(params["start"]) + int(params["count"]))
-            response = requests.get(base_url, params=params)
+            response = requests.get(base_url, params=params, headers=headers)
             if response.status_code != 200:
+                print(f"SEC retrieval error on pagination: {response.text}")
                 break
             feed = response.content
             root = ElementTree.fromstring(feed)