backend: add weather tool

src/community/config/tools.py

@@ -7,6 +7,7 @@
     ConnectorRetriever,
     LlamaIndexUploadPDFRetriever,
     PubMedRetriever,
+    WeatherDataLoader,
 )
 
 
@@ -64,6 +65,15 @@ class CommunityToolName(StrEnum):
         category=Category.Function,
         description="Evaluate arithmetic expressions.",
     ),
+    CommunityToolName.Weather: ManagedTool(
+        name=CommunityToolName.Weather,
+        implementation=WeatherDataLoader,
+        is_visible=True,
+        is_available=WeatherDataLoader.is_available(),
+        error_message="WeatherDataLoader is not available.",
+        category=Category.DataLoader,
+        description="Retrieves weather data.",
+    ),
 }
 
 # For main.py cli setup script

src/community/tests/tools/retrieval/test_weather.py

@@ -0,0 +1,8 @@
+from community.tools.retrieval.weather import WeatherDataLoader
+
+
+def test_pub_med_retriever():
+    retriever = WeatherDataLoader()()
+    result = retriever.retrieve_documents("What is the weather in Toronto?")
+    assert len(result) > 0
+    assert "text" in result[0]

src/community/tools/retrieval/__init__.py

@@ -2,10 +2,12 @@
 from community.tools.retrieval.connector import ConnectorRetriever
 from community.tools.retrieval.llama_index import LlamaIndexUploadPDFRetriever
 from community.tools.retrieval.pub_med import PubMedRetriever
+from community.tools.retrieval.weather import WeatherDataLoader
 
 __all__ = [
     "ArxivRetriever",
     "ConnectorRetriever",
     "LlamaIndexUploadPDFRetriever",
     "PubMedRetriever",
+    "WeatherDataLoader",
 ]

src/community/tools/retrieval/weather.py

@@ -0,0 +1,18 @@
+from typing import Any, Dict, List
+
+from langchain_community.document_loaders.weather import WeatherDataLoader
+
+from community.tools import BaseRetrieval
+
+
+class WeatherDataLoader(BaseRetrieval):
+    def __init__(self):
+        self.client = WeatherDataLoader()
+
+    @classmethod
+    def is_available(cls) -> bool:
+        return True
+
+    def retrieve_documents(self, query: str, **kwargs: Any) -> List[Dict[str, Any]]:
+        result = self.client.from_params(query).load()
+        return [{"text": result}]