Update tools/code_search_tool.py

tools/code_search_tool.py

@@ -1,9 +1,9 @@
-from crewai.tools import BaseTool
+from langchain.tools import BaseTool
 
 
 class CodeSearchTool(BaseTool):
-    name: str = "Code Search Tool"
-    description: str = "Searches relevant code snippets from vector database"
+    name = "Code Search Tool"
+    description = "Searches relevant code snippets"
 
     def __init__(self, vector_db):
         super().__init__()
@@ -11,4 +11,7 @@ class CodeSearchTool(BaseTool):
 
     def _run(self, query: str):
         docs = self.vector_db.similarity_search(query, k=5)
-        return "\n\n".join([doc.page_content for doc in docs])
+
+        return "\n\n".join(
+            [doc.page_content for doc in docs]
+        )