Update tools.py

tools.py

@@ -269,4 +269,24 @@ def create_video_from_images_and_audio(images, speeches, zoom_factor=1.2):
     #     except Exception as e:
     #         print(f"Error removing file {temp_file}: {e}")
     
-    return "./final_video.mp4"
+    return "./final_video.mp4"
+
+class WikiInputs(BaseModel):
+    """Inputs to the wikipedia tool."""
+    query: str = Field(description="query to look up in Wikipedia, should be 3 or less words")
+
+api_wrapper = WikipediaAPIWrapper(top_k_results=3)#, doc_content_chars_max=100)
+
+wiki_tool = WikipediaQueryRun(
+    name="wiki-tool",
+    description="{query:'input here'}",
+    args_schema=WikiInputs,
+    api_wrapper=api_wrapper,
+    return_direct=True,
+)
+
+wiki = Tool(
+    name = 'wikipedia',
+    func = wiki_tool.run,
+    description= "{query:'input here'}"
+)