ws-bom-robot-app 0.0.59__py3-none-any.whl → 0.0.60__py3-none-any.whl

ws_bom_robot_app/llm/tools/tool_manager.py

@@ -171,7 +171,46 @@ class ToolManager:
                 final_results.append({"url": url, "content": "Page not found"})
         return final_results
 
-
+    async def search_online_google(self, query: str):
+      from langchain_google_community import GoogleSearchAPIWrapper
+      from ws_bom_robot_app.llm.tools.utils import fetch_page, extract_content_with_trafilatura
+      import aiohttp, asyncio
+      secrets = {}
+      for d in self.app_tool.secrets:
+        secrets[d.get("secretId")] = d.get("secretValue")
+      search_type = secrets.get("searchType")
+      if search_type:
+          search_kwargs = {"searchType" : search_type}
+      search = GoogleSearchAPIWrapper(
+          google_api_key=secrets.get("GOOGLE_API_KEY"),
+          google_cse_id=secrets.get("GOOGLE_CSE_ID"),
+      )
+      if search_type:
+          raw_results = search.results(query=query,
+                     num_results=secrets.get("num_results", 5),
+                     search_params=search_kwargs)
+          return raw_results
+      raw_results = search.results(
+          query=query,
+          num_results=secrets.get("num_results", 5)
+      )
+      urls = [r["link"] for r in raw_results]
+      async with aiohttp.ClientSession() as session:
+          tasks = [fetch_page(session, url) for url in urls]
+          responses = await asyncio.gather(*tasks)
+      final_results = []
+      for item in responses:
+          url = item["url"]
+          html = item["html"]
+          if html:
+              content = await extract_content_with_trafilatura(html)
+              if content:
+                  final_results.append({"url": url, "content": content, "type": "web"})
+              else:
+                  final_results.append({"url": url, "content": "No content found", "type": "web"})
+          else:
+              final_results.append({"url": url, "content": "Page not found", "type": "web"})
+      return final_results
     #endregion
 
     #class variables (static)
@@ -180,6 +219,7 @@ class ToolManager:
         "image_generator": ToolConfig(function=image_generator, model=ImageGeneratorInput),
         "llm_chain": ToolConfig(function=llm_chain, model=LlmChainInput),
         "search_online": ToolConfig(function=search_online, model=SearchOnlineInput),
+        "search_online_google": ToolConfig(function=search_online_google, model=SearchOnlineInput),
     }
 
     #instance methods

{ws_bom_robot_app-0.0.59.dist-info → ws_bom_robot_app-0.0.60.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ws_bom_robot_app
-Version: 0.0.59
+Version: 0.0.60
 Summary: A FastAPI application serving ws bom/robot/llm platform ai.
 Home-page: https://github.com/websolutespa/bom
 Author: Websolute Spa
@@ -50,6 +50,7 @@ Requires-Dist: unstructured-ingest[slack]
 Requires-Dist: html5lib==1.1
 Requires-Dist: markdownify==0.14.1
 Requires-Dist: duckduckgo-search==8.0.4
+Requires-Dist: langchain_google_community==2.0.7
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{ws_bom_robot_app-0.0.59.dist-info → ws_bom_robot_app-0.0.60.dist-info}/RECORD

@@ -26,7 +26,7 @@ ws_bom_robot_app/llm/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 ws_bom_robot_app/llm/providers/llm_manager.py,sha256=zIkxgTLYQCcup2Ixf4eWap4mNinuJH2YmkjLjZGDyJM,8371
 ws_bom_robot_app/llm/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/llm/tools/tool_builder.py,sha256=p_Q32_-OSydcxzj69PgPIuiny816zYv5dVsCHSY0ELc,1188
-ws_bom_robot_app/llm/tools/tool_manager.py,sha256=adWvaSIur5Ez2gGsuTFqNprZZlQP6ZZj5WknzJjtQ0c,8355
+ws_bom_robot_app/llm/tools/tool_manager.py,sha256=J8WQnjXpLe3Sv6thUi_kgXIDBmt1Z4C-ASzs4RHsVNg,10253
 ws_bom_robot_app/llm/tools/utils.py,sha256=Ba7ScFZPVJ3ke8KLO8ik1wyR2f_zC99Bikqx0OGnKoI,1924
 ws_bom_robot_app/llm/tools/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/llm/tools/models/main.py,sha256=pBQNWPd1OZgZ2xkOnUOawNbujQ5oJXLdyuAex1afLWc,579
@@ -65,7 +65,7 @@ ws_bom_robot_app/llm/vector_store/loader/__init__.py,sha256=47DEQpj8HBSa-_TImW-5
 ws_bom_robot_app/llm/vector_store/loader/base.py,sha256=L_ugekNuAq0N9O-24wtlHSNHkqSeD-KsJrfGt_FX9Oc,5340
 ws_bom_robot_app/llm/vector_store/loader/docling.py,sha256=yP0zgXLeFAlByaYuj-6cYariuknckrFds0dxdRcnVz8,3456
 ws_bom_robot_app/llm/vector_store/loader/json_loader.py,sha256=LDppW0ZATo4_1hh-KlsAM3TLawBvwBxva_a7k5Oz1sc,858
-ws_bom_robot_app-0.0.59.dist-info/METADATA,sha256=hTSpZWSUDrn0IY9jsYvUCGocnzK9kbrEiRINEwiDVe0,8406
-ws_bom_robot_app-0.0.59.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ws_bom_robot_app-0.0.59.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
-ws_bom_robot_app-0.0.59.dist-info/RECORD,,
+ws_bom_robot_app-0.0.60.dist-info/METADATA,sha256=qOMWEMQar7dYSZllRX9OOj_F198nzb4Xmd6Ur3MR1FE,8456
+ws_bom_robot_app-0.0.60.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ws_bom_robot_app-0.0.60.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
+ws_bom_robot_app-0.0.60.dist-info/RECORD,,