From eaf2738b4d78165f8190efe84fbcd65ac5558844 Mon Sep 17 00:00:00 2001 From: Joe G Date: Thu, 6 Feb 2025 17:23:51 -0800 Subject: [PATCH] Fix build issues after refactor prompts --- pyproject.toml | 5 ++++- src/smolagents/__init__.py | 1 - 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 3f49945fc..30e28179e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -99,6 +99,9 @@ lint.select = ["E", "F", "I", "W"] known-first-party = ["smolagents"] lines-after-imports = 2 +[tool.setuptools.package-data] +"smolagents" = ["prompts/*"] + [project.scripts] smolagent = "smolagents.cli:main" -webagent = "smolagents.vision_web_browser:main" \ No newline at end of file +webagent = "smolagents.vision_web_browser:main" diff --git a/src/smolagents/__init__.py b/src/smolagents/__init__.py index 37a545ca4..a2aa13710 100644 --- a/src/smolagents/__init__.py +++ b/src/smolagents/__init__.py @@ -25,7 +25,6 @@ from .memory import * from .models import * from .monitoring import * -from .prompts import * from .tools import * from .utils import * from .cli import *