chore: sistema de imports

openiziai/__init__.py

@@ -2,6 +2,6 @@
 
 __version__ = '0.1.0'
 
-from .agents import Agent as Agent
-from .fine_tuning import FineTuning as FineTuning
+from . import agents as agents
+from . import tools as tools
 from .task import Task as Task

openiziai/agents/__init__.py

@@ -1,5 +1,6 @@
 """Módulo que gerencia métodos relacionado aos Agentes."""
 
 from .agent import Agent
+from .manager import AgentManager
 
-__all__ = ['Agent']
+__all__ = ['Agent', 'AgentManager']

openiziai/fine_tuning.py

@@ -126,7 +126,6 @@ def status(self) -> Optional[str]:
 
         if self._job_status != JobStatus.COMPLETED:
             _job_status = self.client.fine_tuning.jobs.retrieve(job_id).status
-            print(_job_status)
             self._job_status = JobStatus(_job_status)
 
         return self._job_status.name

tests/agents/test_agent.py

@@ -1,7 +1,7 @@
 import pytest
 from pydantic import ValidationError
 
-from openiziai.agents.agent import Agent
+from openiziai.agents import Agent
 from openiziai.schemas import GPTModel, Message
 
 
@@ -50,7 +50,6 @@ def test_agent_prompt(openai_chat, valid_task):
     expected_temperature = 0.5
 
     result = agent.prompt(prompt='teste')
-    print(result.response)
 
     assert result.id == '123'
     assert result.prompt == 'teste'

tests/agents/test_manager.py

@@ -1,6 +1,6 @@
 import pytest
 
-from openiziai.agents.manager import AgentManager
+from openiziai.agents import AgentManager
 from openiziai.schemas import Message
 
 

tests/tools/test_prep_data.py

@@ -1,7 +1,7 @@
 import shutil
 from pathlib import Path
 
-from openiziai.tools.prep_data import prep_data
+from openiziai.tools import prep_data
 
 
 def test_prep_data_with_persistant_data():

tests/tools/test_train_data.py

@@ -6,7 +6,7 @@
 import pytest
 from pydantic import ValidationError
 
-from openiziai.tools.train_data import TrainDataTool
+from openiziai.tools import TrainDataTool
 
 
 @pytest.mark.trio()