added smolagents example

ScaDS · Dec 30, 2024 · 9a9be64 · 9a9be64
1 parent 6df9cc8
commit 9a9be64
Show file tree

Hide file tree

Showing 5 changed files with 669 additions and 3 deletions.
diff --git a/docs/00_setup/environment-cpu.yml b/docs/00_setup/environment-cpu.yml
@@ -43,4 +43,5 @@ dependencies:
       - llama-index-embeddings-huggingface
       - tokenizers==0.21.0
       - pypdf
+      - smolagents==0.1.3
 prefix: C:\Users\haase\miniconda3\envs\genai-cpu
diff --git a/docs/00_setup/environment-gpu.yml b/docs/00_setup/environment-gpu.yml
@@ -49,4 +49,5 @@ dependencies:
       - llama-index-embeddings-huggingface
       - tokenizers==0.21.0
       - pypdf
+      - smolagents==0.1.3
 prefix: C:\Users\haase\miniconda3\envs\genai-gpu
diff --git a/docs/35_agents/agent_tools.py b/docs/35_agents/agent_tools.py
@@ -15,4 +15,24 @@ def print_messages(message_list):
                 elif isinstance(c, autogen_core.models._types.FunctionExecutionResult):
                     display(Markdown(f" = {c.content}\n\n"))    
                 else:
-                    display(Markdown(f"'{c}' \n\n"))
+                    display(Markdown(f"'{c}' \n\n"))
+
+
+
+def get_arxiv_metadata(arxiv_id):
+    import arxiv
+    search = arxiv.Search(id_list=[arxiv_id])
+    paper = next(search.results())
+
+    metadata = {
+        'title': paper.title,
+        'authors': [author.name for author in paper.authors],
+        'published': paper.published,
+        'summary': paper.summary,
+        'doi': paper.doi,
+        'primary_category': paper.primary_category,
+        'categories': paper.categories,
+        'pdf_url': paper.pdf_url
+    }
+    return metadata
+