Create code_search_tool.py

tools/code_search_tool.py

@@ -0,0 +1,14 @@
+from crewai.tools import BaseTool
+
+
+class CodeSearchTool(BaseTool):
+    name: str = "Code Search Tool"
+    description: str = "Searches relevant code snippets from vector database"
+
+    def __init__(self, vector_db):
+        super().__init__()
+        self.vector_db = vector_db
+
+    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])