✨ feat: add support for Ollama and Transformers models in app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel, load_tool, tool, LiteLLMModel
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel, load_tool, tool, LiteLLMModel, TransformersModel
 import datetime
 import requests
 import pytz
@@ -56,6 +56,24 @@ model = LiteLLMModel(
   api_key=os.getenv("LITELLM_API_KEY")
 )
 
+# ollama
+# model = LiteLLMModel(
+#   model_id="ollama_chat/deepseek-r1:7b",
+#   max_tokens=2096,
+#   temperature=0.6,
+#   api_base="http://localhost:11434",
+#   num_ctx=8192
+# )
+
+# transformer
+# model = TransformersModel(
+#   model_id="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
+#   device_map="auto",
+#   torch_dtype="auto",
+#   max_new_tokens=2096,
+#   temperature=0.6,
+# )
+
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)