synth-ai 0.1.0.dev39__py3-none-any.whl → 0.1.0.dev50__py3-none-any.whl

tests/test_recursive_structured_outputs.py

@@ -1,180 +0,0 @@
-import asyncio
-import unittest
-from typing import List
-
-from pydantic import BaseModel, Field
-
-from synth_ai.zyk.lms.core.main import LM
-
-
-# Define example structured output models
-class SimpleResponse(BaseModel):
-    message: str
-    confidence: float
-
-
-class ComplexResponse(BaseModel):
-    title: str
-    tags: List[str]
-    content: str
-
-
-class NestedResponse(BaseModel):
-    main_category: str
-    subcategories: List[str]
-    details: SimpleResponse
-
-
-# Define nested structured output models
-class Address(BaseModel):
-    street: str
-    city: str
-    country: str
-
-
-class PersonalInfo(BaseModel):
-    name: str
-    age: int
-    address: Address
-
-
-class WorkInfo(BaseModel):
-    company: str
-    position: str
-    years_experience: int
-
-
-class NestedPersonResponse(BaseModel):
-    personal: PersonalInfo
-    work: WorkInfo
-    skills: List[str]
-
-
-class ProjectDetails(BaseModel):
-    name: str
-    description: str
-    technologies: List[str]
-
-
-class NestedPortfolioResponse(BaseModel):
-    developer: PersonalInfo
-    projects: List[ProjectDetails]
-    total_experience: int
-
-
-class NestedCompanyResponse(BaseModel):
-    name: str
-    founded: int
-    headquarters: Address
-    employees: List[PersonalInfo]
-    main_products: List[str]
-
-
-class TestLMStructuredOutputs(unittest.TestCase):
-    @classmethod
-    def setUpClass(cls):
-        # Initialize the LM once for all tests
-        cls.lm = LM(
-            model_name="gpt-4o-mini",
-            formatting_model_name="gpt-4o-mini",
-            temperature=0.7,
-            max_retries="Few",
-            structured_output_mode="forced_json",
-        )
-
-    def test_sync_simple_response(self):
-        result = self.lm.respond_sync(
-            system_message="You are a helpful assistant.",
-            user_message="Give me a short greeting and your confidence level.",
-            response_model=SimpleResponse,
-        )
-        self.assertIsInstance(result, SimpleResponse)
-        self.assertIsInstance(result.message, str)
-        self.assertIsInstance(result.confidence, float)
-        self.assertGreaterEqual(result.confidence, 0)
-        self.assertLessEqual(result.confidence, 1)
-
-    def test_sync_complex_response(self):
-        result = self.lm.respond_sync(
-            system_message="You are a content creator.",
-            user_message="Create a short blog post about AI.",
-            response_model=ComplexResponse,
-        )
-        self.assertIsInstance(result, ComplexResponse)
-        self.assertIsInstance(result.title, str)
-        self.assertIsInstance(result.tags, list)
-        self.assertIsInstance(result.content, str)
-
-    async def async_nested_response(self):
-        result = await self.lm.respond_async(
-            system_message="You are a categorization expert.",
-            user_message="Categorize 'Python' and provide a brief description.",
-            response_model=NestedResponse,
-        )
-        self.assertIsInstance(result, NestedResponse)
-        self.assertIsInstance(result.main_category, str)
-        self.assertIsInstance(result.subcategories, list)
-        self.assertIsInstance(result.details, SimpleResponse)
-
-    def test_async_nested_response(self):
-        asyncio.run(self.async_nested_response())
-
-
-class TestLMNestedStructuredOutputs(unittest.TestCase):
-    @classmethod
-    def setUpClass(cls):
-        # Initialize the LM once for all tests
-        cls.lm = LM(
-            model_name="gpt-4o-mini",
-            formatting_model_name="gpt-4o-mini",
-            temperature=0.7,
-            max_retries="Few",
-            structured_output_mode="forced_json",
-        )
-
-    def test_sync_nested_person_response(self):
-        result = self.lm.respond_sync(
-            system_message="You are an HR assistant.",
-            user_message="Provide detailed information about a fictional employee named John Doe.",
-            response_model=NestedPersonResponse,
-        )
-        self.assertIsInstance(result, NestedPersonResponse)
-        self.assertIsInstance(result.personal, PersonalInfo)
-        self.assertIsInstance(result.personal.address, Address)
-        self.assertIsInstance(result.work, WorkInfo)
-        self.assertIsInstance(result.skills, list)
-
-    def test_sync_nested_portfolio_response(self):
-        result = self.lm.respond_sync(
-            system_message="You are a portfolio manager.",
-            user_message="Create a portfolio for a fictional software developer with multiple projects.",
-            response_model=NestedPortfolioResponse,
-        )
-        self.assertIsInstance(result, NestedPortfolioResponse)
-        self.assertIsInstance(result.developer, PersonalInfo)
-        self.assertIsInstance(result.developer.address, Address)
-        self.assertIsInstance(result.projects, list)
-        for project in result.projects:
-            self.assertIsInstance(project, ProjectDetails)
-        self.assertIsInstance(result.total_experience, int)
-
-    async def async_nested_company_response(self):
-        result = await self.lm.respond_async(
-            system_message="You are a company information specialist.",
-            user_message="Provide detailed information about a fictional tech company.",
-            response_model=NestedCompanyResponse,
-        )
-        self.assertIsInstance(result, NestedCompanyResponse)
-        self.assertIsInstance(result.headquarters, Address)
-        self.assertIsInstance(result.employees, list)
-        for employee in result.employees:
-            self.assertIsInstance(employee, PersonalInfo)
-            self.assertIsInstance(employee.address, Address)
-        self.assertIsInstance(result.main_products, list)
-
-    def test_async_nested_company_response(self):
-        asyncio.run(self.async_nested_company_response())
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/test_structured_outputs.py

@@ -1,100 +0,0 @@
-import asyncio
-import unittest
-from typing import List
-
-from pydantic import BaseModel, Field
-
-from synth_ai.zyk.lms.core.main import LM
-
-
-# Define example structured output models
-class SimpleResponse(BaseModel):
-    message: str
-    confidence_between_zero_one: float = Field(
-        ..., description="Confidence level between 0 and 1"
-    )
-
-
-class ComplexResponse(BaseModel):
-    title: str
-    tags: List[str]
-    content: str
-
-
-class NestedResponse(BaseModel):
-    main_category: str
-    subcategories: List[str]
-    details: SimpleResponse
-
-
-class TestLMStructuredOutputs(unittest.TestCase):
-    @classmethod
-    def setUpClass(cls):
-        # Initialize LMs for both forced_json and stringified_json modes
-        cls.lm_forced_json = LM(
-            model_name="gpt-4o-mini",
-            formatting_model_name="gpt-4o-mini",
-            temperature=0.7,
-            max_retries="Few",
-            structured_output_mode="forced_json",
-        )
-        cls.lm_stringified_json = LM(
-            model_name="gpt-4o-mini",
-            formatting_model_name="gpt-4o-mini",
-            temperature=0.7,
-            max_retries="Few",
-            structured_output_mode="stringified_json",
-        )
-
-    def test_sync_simple_response(self):
-        for lm in [self.lm_forced_json, self.lm_stringified_json]:
-            with self.subTest(
-                mode=lm.structured_output_handler.handler.structured_output_mode
-            ):
-                result = lm.respond_sync(
-                    system_message="You are a helpful assistant.",
-                    user_message="Give me a short greeting and your confidence level.",
-                    response_model=SimpleResponse,
-                )
-                self.assertIsInstance(result, SimpleResponse)
-                self.assertIsInstance(result.message, str)
-                self.assertIsInstance(result.confidence_between_zero_one, float)
-                self.assertGreaterEqual(result.confidence_between_zero_one, 0)
-                self.assertLessEqual(result.confidence_between_zero_one, 1)
-
-    def test_sync_complex_response(self):
-        for lm in [self.lm_forced_json, self.lm_stringified_json]:
-            with self.subTest(
-                mode=lm.structured_output_handler.handler.structured_output_mode
-            ):
-                result = lm.respond_sync(
-                    system_message="You are a content creator.",
-                    user_message="Create a short blog post about AI.",
-                    response_model=ComplexResponse,
-                )
-                self.assertIsInstance(result, ComplexResponse)
-                self.assertIsInstance(result.title, str)
-                self.assertIsInstance(result.tags, list)
-                self.assertIsInstance(result.content, str)
-
-    async def async_nested_response(self, lm):
-        result = await lm.respond_async(
-            system_message="You are a categorization expert.",
-            user_message="Categorize 'Python' and provide a brief description.",
-            response_model=NestedResponse,
-        )
-        self.assertIsInstance(result, NestedResponse)
-        self.assertIsInstance(result.main_category, str)
-        self.assertIsInstance(result.subcategories, list)
-        self.assertIsInstance(result.details, SimpleResponse)
-
-    def test_async_nested_response(self):
-        for lm in [self.lm_forced_json, self.lm_stringified_json]:  #
-            with self.subTest(
-                mode=lm.structured_output_handler.handler.structured_output_mode
-            ):
-                asyncio.run(self.async_nested_response(lm))
-
-
-if __name__ == "__main__":
-    unittest.main()