version bump and minor bugfixes

agent.py

@@ -21,35 +21,43 @@ citation_description = '''
     Citation must include the volume number, reporter, and first page. For example: 253 P.2d 136.
 '''
 
-def extract_components_from_citation(citation: str) -> Tuple[int, str, int]:
+def extract_components_from_citation(citation: str) -> dict:
     citation_components = citation.split(' ')
     volume_num = citation_components[0]
     reporter = '-'.join(citation_components[1:-1]).replace('.', '').lower()
     first_page = citation_components[-1]
 
     if not volume_num.isdigit():
-        raise ValueError("volume number must be a number.")
+        return {}
     if not first_page.isdigit():
-        raise ValueError("first page number must be a number.")
+        return {}
 
-    return int(volume_num), reporter, int(first_page)
+    return {'volume': int(volume_num), 'reporter': reporter, 'first_page': int(first_page)}
 
 def create_assistant_tools(cfg):
 
     def get_opinion_text(
-            case_citation = Field(description = citation_description),
-            summarize: Optional[bool] = False
+            case_citation: str = Field(description = citation_description),
+            summarize: bool = Field(default=True, description="if True returns case summary, otherwise the full text of the case")
             ) -> str:
         """
-        Given case citation, returns the full opinion/ruling text of the case.
-        if summarize is True, the text is summarized.
+        Returns the full opinion/ruling text of the case, or the summary if summarize=True.
         If there is more than one opinion for the case, the type of each opinion is returned with the text, 
         and the opinions (or their summaries) are separated by semicolons (;)
+
+        Args
+            case_citation (str): the citation for a particular case. Citation must include the volume number, reporter, and first page. For example: 253 P.2d 136.
+            summarize (bool): True to return just a summary of the case, False to return full case text.
         """
-        volume_num, reporter, first_page = extract_components_from_citation(case_citation)
+        citation_dict = extract_components_from_citation(case_citation)
+        if not citation_dict:
+            return f"Citation is invalid: {case_citation}."
+        reporter = citation_dict['reporter']
+        volume_num = citation_dict['volume']
+        first_page = citation_dict['first_page']
         response = requests.get(f"https://static.case.law/{reporter}/{volume_num}/cases/{first_page:04d}-01.json")
         if response.status_code != 200:
-            return "Case not found; please check the citation."
+            return f"Case not found; please check the citation {case_citation}."
         res = json.loads(response.text)
 
         if len(res["casebody"]["opinions"]) == 1:
@@ -69,10 +77,15 @@ def create_assistant_tools(cfg):
         """
         Given a case citation, returns a valid web url to a pdf of the case record
         """
-        volume_num, reporter, first_page = extract_components_from_citation(case_citation)
+        citation_dict = extract_components_from_citation(case_citation)
+        if not citation_dict:
+            return f"Citation is invalid: {case_citation}."
+        reporter = citation_dict['reporter']
+        volume_num = citation_dict['volume']
+        first_page = citation_dict['first_page']
         response = requests.get(f"https://static.case.law/{reporter}/{volume_num}/cases/{first_page:04d}-01.json")
         if response.status_code != 200:
-            return "Case not found; please check the citation."
+            return f"Case not found; please check the citation {case_citation}."
         res = json.loads(response.text)
         page_number = res["first_page_order"]
         return f"https://static.case.law/{reporter}/{volume_num}.pdf#page={page_number}"
@@ -83,7 +96,12 @@ def create_assistant_tools(cfg):
         """
         Given a case citation, returns a valid web url to a page with information about the case.
         """
-        volume_num, reporter, first_page = extract_components_from_citation(case_citation)
+        citation_dict = extract_components_from_citation(case_citation)
+        if not citation_dict:
+            return f"Citation is invalid: {case_citation}."
+        reporter = citation_dict['reporter']
+        volume_num = citation_dict['volume']
+        first_page = citation_dict['first_page']
         url = f"https://case.law/caselaw/?reporter={reporter}&volume={volume_num}&case={first_page:04d}-01"
         response = requests.get(url)
         if response.status_code != 200:
@@ -96,7 +114,12 @@ def create_assistant_tools(cfg):
         """
         Given a case citation, returns its name and name abbreviation.
         """
-        volume_num, reporter, first_page = extract_components_from_citation(case_citation)
+        citation_dict = extract_components_from_citation(case_citation)
+        if not citation_dict:
+            return f"Citation is invalid: {case_citation}.", f"Citation is invalid: {case_citation}."
+        reporter = citation_dict['reporter']
+        volume_num = citation_dict['volume']
+        first_page = citation_dict['first_page']
         response = requests.get(f"https://static.case.law/{reporter}/{volume_num}/cases/{first_page:04d}-01.json")
         if response.status_code != 200:
             return "Case not found", "Case not found"
@@ -110,7 +133,12 @@ def create_assistant_tools(cfg):
         Given a case citation, returns a list of cases that are cited by the opinion of this case.
         The output is a list of cases, each a dict with the citation, name and name_abbreviation of the case.
         """
-        volume_num, reporter, first_page = extract_components_from_citation(case_citation)
+        citation_dict = extract_components_from_citation(case_citation)
+        if not citation_dict:
+            return [f"Citation is invalid: {case_citation}."]
+        reporter = citation_dict['reporter']
+        volume_num = citation_dict['volume']
+        first_page = citation_dict['first_page']
         response = requests.get(f"https://static.case.law/{reporter}/{volume_num}/cases/{first_page:04d}-01.json")
         if response.status_code != 200:
             return "Case not found; please check the citation."
@@ -159,7 +187,7 @@ def create_assistant_tools(cfg):
         reranker = "multilingual_reranker_v1", rerank_k = 100, 
         n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
         summary_num_results = 10,
-        vectara_summarizer = 'vectara-summary-ext-24-05-med-omni',
+        vectara_summarizer = 'vectara-experimental-summary-ext-2023-12-11-med-omni',
         include_citations = False,
     )
 

requirements.txt

@@ -6,4 +6,4 @@ streamlit-feedback==0.1.3
 uuid==1.30
 langdetect==1.0.9
 langcodes==3.4.0
-vectara-agentic==0.1.13
+vectara-agentic==0.1.15