setup gemini code

.gitignore

@@ -0,0 +1,51 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+build/
+dist/
+*.egg-info/
+.eggs/
+
+# Virtual environments
+.venv/
+venv/
+env/
+ENV/
+env.bak/
+venv.bak/
+
+# uv environment files
+.uv/
+
+# IDE/editor folders
+.vscode/
+.idea/
+*.sublime-project
+*.sublime-workspace
+
+# Jupyter Notebook checkpoints
+.ipynb_checkpoints/
+
+# macOS files
+.DS_Store
+
+# Logs and local data
+*.log
+*.sqlite3
+*.db
+
+# Environment variables
+.env
+.env.*
+.local.env
+
+# GRPC plugin artifacts
+*.proto.ts
+
+.gradio/

.python-version

@@ -0,0 +1 @@
+3.12

app.py

@@ -1,13 +1,22 @@
 import gradio as gr
 import random
-from smolagents import GradioUI, CodeAgent, HfApiModel
+import os
+from smolagents import GradioUI, CodeAgent, HfApiModel, LiteLLMModel, ApiModel
 
 # Import our custom tools from their modules
 from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
 from retriever import load_guest_dataset
 
 # Initialize the Hugging Face model
-model = HfApiModel()
+#model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
+# model = LiteLLMModel(model_id="gpt-4")
+
+# Use Gemini 1.5 Flash - Long context (1M tokens) and cheap
+model = LiteLLMModel(
+    model_id="gemini/gemini-1.5-flash",
+    api_key=os.getenv("GEMINI_API_KEY"),
+    max_tokens=2000  # Can be higher due to long context window
+)
 
 # Initialize the web search tool
 search_tool = DuckDuckGoSearchTool()
@@ -21,12 +30,12 @@ hub_stats_tool = HubStatsTool()
 # Load the guest dataset and initialize the guest info tool
 guest_info_tool = load_guest_dataset()
 
-# Create Alfred with all the tools
+# Create Alfred with full configuration - Gemini can handle more tokens
 alfred = CodeAgent(
     tools=[guest_info_tool, weather_info_tool, hub_stats_tool, search_tool], 
     model=model,
-    add_base_tools=True,  # Add any additional base tools
-    planning_interval=3   # Enable planning every 3 steps
+    add_base_tools=True,  # Re-enable base tools
+    planning_interval=3   # Re-enable planning with shorter interval
 )
 
 if __name__ == "__main__":

main.py

@@ -0,0 +1,6 @@
+def main():
+    print("Hello from agentic-rag!")
+
+
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -0,0 +1,16 @@
+[project]
+name = "agentic-rag"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "datasets>=3.6.0",
+    "duckduckgo-search>=8.0.4",
+    "gradio>=5.35.0",
+    "langchain-community>=0.3.26",
+    "openai>=1.93.0",
+    "rank-bm25>=0.2.2",
+    "smolagents[litellm]==1.18.0",
+    "together>=1.5.17",
+]

retriever.py

@@ -47,7 +47,4 @@ def load_guest_dataset():
     ]
 
     # Return the tool
-    return GuestInfoRetrieverTool(docs)
-
-
-
+    return GuestInfoRetrieverTool(docs)

tools.py

@@ -5,7 +5,7 @@ from huggingface_hub import list_models
 
 
 # Initialize the DuckDuckGo search tool
-#search_tool = DuckDuckGoSearchTool()
+search_tool = DuckDuckGoSearchTool()
 
 
 class WeatherInfoTool(Tool):