From 008252cddee67c17e2ed0b278880c65b294f32c7 Mon Sep 17 00:00:00 2001
From: Maciej Majek <maciej.majek@robotec.ai>
Date: Thu, 6 Feb 2025 20:25:26 +0100
Subject: [PATCH] fix: tests

---
 poetry.lock                     | 66 ++++++++++++++++++++++++++++++++-
 pyproject.toml                  | 16 ++++----
 src/rai_asr/rai_asr/__init__.py |  4 ++
 src/rai_core/README.md          |  0
 src/rai_core/rai/cli/rai_cli.py |  2 +-
 src/rai_tts/README.md           |  0
 6 files changed, 78 insertions(+), 10 deletions(-)
 create mode 100644 src/rai_core/README.md
 create mode 100644 src/rai_tts/README.md

diff --git a/poetry.lock b/poetry.lock
index d6da53572..93059b345 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -5653,6 +5653,70 @@ files = [
 [package.dependencies]
 cffi = {version = "*", markers = "implementation_name == \"pypy\""}
 
+[[package]]
+name = "rai"
+version = "1.0.0"
+description = "Core functionality for RAI framework"
+optional = false
+python-versions = "^3.10, <3.13"
+files = []
+develop = true
+
+[package.dependencies]
+coloredlogs = "^15.0.1"
+deprecated = "^1.2.14"
+langchain = "*"
+langchain-core = "^0.3"
+langgraph = "*"
+markdown = "^3.6"
+requests = "^2.32.2"
+rich = "^13.7.1"
+tomli = "^2.0.1"
+tomli-w = "^1.1.0"
+tqdm = "^4.66.4"
+
+[package.source]
+type = "directory"
+url = "src/rai_core"
+
+[[package]]
+name = "rai-asr"
+version = "1.0.0"
+description = "Automatic Speech Recognition module for RAI framework"
+optional = false
+python-versions = "^3.10, <3.13"
+files = []
+develop = true
+
+[package.dependencies]
+faster-whisper = "^1.1.1"
+openai-whisper = "^20231117"
+pydub = "^0.25.1"
+scipy = "^1.14.0"
+sounddevice = "^0.4.7"
+torchaudio = "^2.3.1"
+
+[package.source]
+type = "directory"
+url = "src/rai_asr"
+
+[[package]]
+name = "rai-tts"
+version = "1.0.0"
+description = "Text-to-Speech module for RAI framework"
+optional = false
+python-versions = "^3.10, <3.13"
+files = []
+develop = true
+
+[package.dependencies]
+elevenlabs = "^1.4.1"
+sounddevice = "^0.4.7"
+
+[package.source]
+type = "directory"
+url = "src/rai_tts"
+
 [[package]]
 name = "ray"
 version = "2.41.0"
@@ -8238,4 +8302,4 @@ cffi = ["cffi (>=1.11)"]
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.10, <3.13"
-content-hash = "158877f96c27f3c9beb75aff8a9608a41b83740afe17ceda2a471378c4bb3545"
+content-hash = "242e440c4ce4b31fa629d198a3d79b0854e84284d013d819a4b7a24e633a1706"
diff --git a/pyproject.toml b/pyproject.toml
index 0705512b8..0d3627c83 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,15 +9,15 @@ classifiers = [
     "Development Status :: 4 - Beta",
     "License :: OSI Approved :: Apache Software License",
 ]
-
-packages = [
-    { include = "rai_core", from = "src" },
-    { include = "rai_asr", from = "src" },
-    { include = "rai_tts", from = "src" },
-]
+package-mode = false
 
 [tool.poetry.dependencies]
 python = "^3.10, <3.13"
+
+rai = {path = "src/rai_core", develop = true}
+rai_asr = {path = "src/rai_asr", develop = true}
+rai_tts = {path = "src/rai_tts", develop = true}
+
 langchain-core = "^0.3"
 langchain = "*"
 langgraph = "*"
@@ -84,8 +84,8 @@ visualnav_transformer = { git = "https://github.com/RobotecAI/visualnav-transfor
 gdown = "^5.2.0"
 
 [build-system]
-requires = ["setuptools>=42", "wheel"]
-build-backend = "setuptools.build_meta"
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"
 
 [tool.isort]
 profile = "black"
diff --git a/src/rai_asr/rai_asr/__init__.py b/src/rai_asr/rai_asr/__init__.py
index ef74fc891..499007aa9 100644
--- a/src/rai_asr/rai_asr/__init__.py
+++ b/src/rai_asr/rai_asr/__init__.py
@@ -11,3 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+"""RAI ASR package."""
+
+__version__ = "0.1.0"
diff --git a/src/rai_core/README.md b/src/rai_core/README.md
new file mode 100644
index 000000000..e69de29bb
diff --git a/src/rai_core/rai/cli/rai_cli.py b/src/rai_core/rai/cli/rai_cli.py
index 4df87cd17..70afb99cf 100644
--- a/src/rai_core/rai/cli/rai_cli.py
+++ b/src/rai_core/rai/cli/rai_cli.py
@@ -194,7 +194,7 @@ def create_rai_ws():
     (package_path / "generated" / "robot_constitution.txt").touch()
 
     default_constitution_path = (
-        "src/rai/rai/cli/resources/default_robot_constitution.txt"
+        "src/rai_core/rai/cli/resources/default_robot_constitution.txt"
     )
     with open(default_constitution_path, "r") as file:
         default_constitution = file.read()
diff --git a/src/rai_tts/README.md b/src/rai_tts/README.md
new file mode 100644
index 000000000..e69de29bb