sk agent

metagpt/tools/search_engine.py

@@ -5,15 +5,32 @@
 @Author  : alexanderwu
 @File    : search_engine.py
 """
-from __future__ import annotations
+# from __future__ import annotations
 
 import importlib
 from typing import Callable, Coroutine, Literal, overload
 
+from semantic_kernel.skill_definition import sk_function
+
 from metagpt.config import CONFIG
 from metagpt.tools import SearchEngineType
 
 
+class SkSearchEngine:
+    def __init__(self):
+        self.search_engine = SearchEngine()
+
+    @sk_function(
+        description="searches results from Google. Useful when you need to find short "
+        "and succinct answers about a specific topic. Input should be a search query.",
+        name="searchAsync",
+        input_description="search",
+    )
+    async def run(self, query: str) -> str:
+        result = await self.search_engine.run(query)
+        return result
+
+
 class SearchEngine:
     """Class representing a search engine.
 
@@ -25,6 +42,7 @@ class SearchEngine:
         run_func: The function to run the search.
         engine: The search engine type.
     """
+
     def __init__(
         self,
         engine: SearchEngineType | None = None,
@@ -33,7 +51,7 @@ class SearchEngine:
         engine = engine or CONFIG.search_engine
         if engine == SearchEngineType.SERPAPI_GOOGLE:
             module = "metagpt.tools.search_engine_serpapi"
-            run_func = importlib.import_module(module).SerpAPIWrapper().run            
+            run_func = importlib.import_module(module).SerpAPIWrapper().run
         elif engine == SearchEngineType.SERPER_GOOGLE:
             module = "metagpt.tools.search_engine_serper"
             run_func = importlib.import_module(module).SerperWrapper().run