Update primitives/tools.py

primitives/tools.py

@@ -63,7 +63,7 @@ TYPE_MAPPING = {
 # Available tools
 # -----------------
 
-async def cricket_source(mode: Literal["live", "upcoming"], want: Literal["text", "href"] ) -> str:
+async def cricket_source(mode: Literal["live", "upcoming"], want: Literal["live", "upcoming"]) -> str:
     """Fetches whole html from source extracts html container that contains necessary details
     
     Args:
@@ -97,7 +97,7 @@ async def cricket_source(mode: Literal["live", "upcoming"], want: Literal["text"
         if mode in ['live', 'upcoming'] and want == "text":
             text = container.get_text(separator=" ", strip= True) 
             return json.dumps({f"{mode} details": str(text)})    
-        elif mode == 'live' and want == 'herf':
+        elif mode == 'live' and want == 'href':
             hrefs_list = container.find_all("a", class_ = "cb-text-link cb-mtch-lnks") 
             hrefs_string = ",".join(str(tag) for tag in herfs_list)
             return json.dumps({"hrefs_strings": herfs_string})
@@ -120,7 +120,7 @@ async def fetch_cricket_details(mode: Literal["live", "upcoming"])-> str:
 
 async def live_cricket_scorecard_href()-> str:
     """String of comma separated anchor tags with herf attributes that pointing to live cricket scorecards """
-    response = await cricket_source('live', 'herf')
+    response = await cricket_source('live', 'href')
     return response
 
 
@@ -292,7 +292,7 @@ TOOL_COMPONENTS = {
     },
     "fetch_cricket_details": {
         "is_gradio_api": False, 
-        "inputs": gr.Radio(["live", "upcomming"], label="Mode"),
+        "inputs": gr.Radio(["live", "upcoming"], label="Mode"),
         "outputs": gr.JSON(label= "Details"), 
     },
     "live_cricket_scorecard_href": {