IncludeCPP 4.0.2__py3-none-any.whl → 4.3.0__py3-none-any.whl

includecpp/core/ai_integration.py

@@ -32,8 +32,12 @@ MODELS = {
     'gpt-3.5-turbo': {'context': 16385, 'endpoint': 'gpt-3.5-turbo'},
     'gpt-4-turbo': {'context': 128000, 'endpoint': 'gpt-4-turbo'},
     'gpt-4o': {'context': 128000, 'endpoint': 'gpt-4o'},
+    'gpt-4o-mini': {'context': 128000, 'endpoint': 'gpt-4o-mini'},
     'gpt-5': {'context': 256000, 'endpoint': 'gpt-5'},
     'gpt-5-nano': {'context': 32000, 'endpoint': 'gpt-5-nano'},
+    'o1': {'context': 200000, 'endpoint': 'o1'},
+    'o1-mini': {'context': 128000, 'endpoint': 'o1-mini'},
+    'o1-preview': {'context': 128000, 'endpoint': 'o1-preview'},
 }
 
 DEFAULT_MODEL = 'gpt-5'
@@ -981,18 +985,43 @@ class AIManager:
             'Content-Type': 'application/json'
         }
         token_limit = min(16000, model_info['context'] // 2)
-        data = {
-            'model': model_info['endpoint'],
-            'messages': [
-                {'role': 'system', 'content': system_prompt},
-                {'role': 'user', 'content': user_prompt}
-            ]
-        }
-        if model.startswith('gpt-5'):
-            data['max_completion_tokens'] = token_limit
+
+        # o1 models use different message format (no system message, combined into user)
+        # gpt-5 models use max_completion_tokens instead of max_tokens
+        is_o1_model = model.startswith('o1')
+        is_gpt5_model = model.startswith('gpt-5')
+
+        if is_o1_model:
+            # o1 models: combine system + user into single user message
+            combined_content = f"{system_prompt}\n\n---\n\n{user_prompt}"
+            data = {
+                'model': model_info['endpoint'],
+                'messages': [
+                    {'role': 'user', 'content': combined_content}
+                ],
+                'max_completion_tokens': token_limit
+            }
+            # o1 models don't support temperature
+        elif is_gpt5_model:
+            # gpt-5 models: use max_completion_tokens, no custom temperature
+            data = {
+                'model': model_info['endpoint'],
+                'messages': [
+                    {'role': 'system', 'content': system_prompt},
+                    {'role': 'user', 'content': user_prompt}
+                ],
+                'max_completion_tokens': token_limit
+            }
         else:
-            data['max_tokens'] = token_limit
-            data['temperature'] = temperature
+            data = {
+                'model': model_info['endpoint'],
+                'messages': [
+                    {'role': 'system', 'content': system_prompt},
+                    {'role': 'user', 'content': user_prompt}
+                ],
+                'max_tokens': token_limit,
+                'temperature': temperature
+            }
         try:
             response = requests.post(OPENAI_API_URL, headers=headers, json=data, timeout=timeout)
             if response.status_code == 200:
@@ -1312,6 +1341,10 @@ class AIManager:
         import platform
         import subprocess
 
+        # Early fail-fast checks
+        if not self.config.get('api_key'):
+            return False, 'No API key configured. Run: includecpp ai setup', []
+
         # Build tools list for prompt
         tools_list = '\n'.join([
             f"- {name}: {info['desc']}\n  Format:\n  {info['format']}"
@@ -1371,9 +1404,9 @@ class AIManager:
 
         prompt = self._build_prompt_with_docs(prompt)
 
-        # Temperature and timeout
+        # Temperature and timeout (reduced for faster failure detection)
         temperature = 0.1 if think_three else (0.2 if think_twice else 0.3)
-        timeout = None if think_three else (420 if think_twice else (300 if think else 180))
+        timeout = 180 if think_three else (120 if think_twice else (90 if think else 60))
 
         # Execute with tool loop
         all_changes = []

includecpp/core/cpp_api_extensions.pyi

@@ -0,0 +1,350 @@
+"""Auto-generated type stubs for IncludeCPP module wrappers.
+
+This file enables VSCode IntelliSense autocomplete for C++ modules.
+DO NOT EDIT - Auto-generated by IncludeCPP build system.
+"""
+
+from typing import Any, List, Dict, Optional, Union, Protocol, overload
+
+class Fast_listModuleWrapper(Protocol):
+    """Type hints for fast_list module wrapper (VSCode autocomplete support)."""
+
+    def getInfo(self) -> Dict[str, Any]:
+        """Get fast_list module information."""
+        ...
+
+    class FastList:
+        """C++ class: FastList"""
+
+        def __init__(self) -> None:
+            """Initialize FastList instance"""
+            ...
+
+        def add(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: add"""
+            ...
+
+        def get(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get"""
+            ...
+
+        def size(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: size"""
+            ...
+
+        def clear(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: clear"""
+            ...
+
+
+    def fast_sort(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: fast_sort"""
+        ...
+
+    def fast_reverse(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: fast_reverse"""
+        ...
+
+    def fast_sum(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: fast_sum"""
+        ...
+
+    def fast_max(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: fast_max"""
+        ...
+
+    def fast_min(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: fast_min"""
+        ...
+
+
+
+class Game_mathModuleWrapper(Protocol):
+    """Type hints for game_math module wrapper (VSCode autocomplete support)."""
+
+    def getInfo(self) -> Dict[str, Any]:
+        """Get game_math module information."""
+        ...
+
+    class Vector2D:
+        """C++ class: Vector2D"""
+
+        @overload
+        def __init__(self) -> None: ...
+        @overload
+        def __init__(self, arg0: float, arg1: float) -> None: ...
+
+        def __init__(self, *args: Any, **kwargs: Any) -> None:
+            """Initialize Vector2D instance"""
+            ...
+
+        def add(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: add"""
+            ...
+
+        def dot(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: dot"""
+            ...
+
+        def length(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: length"""
+            ...
+
+        def normalize(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: normalize"""
+            ...
+
+
+    def boxCollision(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: boxCollision"""
+        ...
+
+    def circleCollision(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: circleCollision"""
+        ...
+
+    def distance(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: distance"""
+        ...
+
+    def square(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: square"""
+        ...
+
+
+
+class Solar_systemModuleWrapper(Protocol):
+    """Type hints for solar_system module wrapper (VSCode autocomplete support)."""
+
+    def getInfo(self) -> Dict[str, Any]:
+        """Get solar_system module information."""
+        ...
+
+    class CelestialBody:
+        """C++ class: CelestialBody"""
+
+        def __init__(self) -> None:
+            """Initialize CelestialBody instance"""
+            ...
+
+        def add_trajectory_point(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: add_trajectory_point"""
+            ...
+
+
+    class SolarSystem:
+        """C++ class: SolarSystem"""
+
+        def __init__(self) -> None:
+            """Initialize SolarSystem instance"""
+            ...
+
+        def calculate_angular_momentum(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: calculate_angular_momentum"""
+            ...
+
+        def calculate_total_energy(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: calculate_total_energy"""
+            ...
+
+        def get_body_count(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_body_count"""
+            ...
+
+        def get_distance_from_sun(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_distance_from_sun"""
+            ...
+
+        def get_energy_error(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_energy_error"""
+            ...
+
+        def get_masses(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_masses"""
+            ...
+
+        def get_names(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_names"""
+            ...
+
+        def get_orbital_period(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_orbital_period"""
+            ...
+
+        def get_positions(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_positions"""
+            ...
+
+        def get_positions_au(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_positions_au"""
+            ...
+
+        def get_radii(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_radii"""
+            ...
+
+        def get_simulation_time(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_simulation_time"""
+            ...
+
+        def get_simulation_time_days(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_simulation_time_days"""
+            ...
+
+        def get_simulation_time_years(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_simulation_time_years"""
+            ...
+
+        def get_speed(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_speed"""
+            ...
+
+        def get_step_count(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_step_count"""
+            ...
+
+        def get_total_energy(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_total_energy"""
+            ...
+
+        def get_trajectory(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_trajectory"""
+            ...
+
+        def get_velocities(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: get_velocities"""
+            ...
+
+        def init_real_solar_system(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: init_real_solar_system"""
+            ...
+
+        def simulate(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: simulate"""
+            ...
+
+        def step(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: step"""
+            ...
+
+
+    def get_AU(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: get_AU"""
+        ...
+
+    def get_DAY(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: get_DAY"""
+        ...
+
+    def get_G(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: get_G"""
+        ...
+
+    def get_YEAR(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: get_YEAR"""
+        ...
+
+
+
+class TemplatesModuleWrapper(Protocol):
+    """Type hints for templates module wrapper (VSCode autocomplete support)."""
+
+    def getInfo(self) -> Dict[str, Any]:
+        """Get templates module information."""
+        ...
+
+    class Pointf:
+        """C++ class: Pointf"""
+
+        @overload
+        def __init__(self) -> None: ...
+        @overload
+        def __init__(self, arg0: float, arg1: float) -> None: ...
+
+        def __init__(self, *args: Any, **kwargs: Any) -> None:
+            """Initialize Pointf instance"""
+            ...
+
+        def distanceFromOrigin(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: distanceFromOrigin"""
+            ...
+
+
+    class Pointi:
+        """C++ class: Pointi"""
+
+        @overload
+        def __init__(self) -> None: ...
+        @overload
+        def __init__(self, arg0: int, arg1: int) -> None: ...
+
+        def __init__(self, *args: Any, **kwargs: Any) -> None:
+            """Initialize Pointi instance"""
+            ...
+
+        def distanceFromOrigin(self, *args: Any, **kwargs: Any) -> Any:
+            """C++ method: distanceFromOrigin"""
+            ...
+
+
+    def maximum(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: maximum"""
+        ...
+
+    def sumVector(self, *args: Any, **kwargs: Any) -> Any:
+        """C++ function: sumVector"""
+        ...
+
+
+
+# CppApi with typed include() overloads for each module
+class CppApi:
+    """C++ API Manager with typed module loading.
+
+    The include() method returns a module wrapper with full type hints
+    for VSCode/PyCharm autocomplete support.
+    """
+
+    def __init__(self, project_root: Optional[str] = None, auto_update: bool = True) -> None:
+        """Initialize CppApi.
+
+        Args:
+            project_root: Path to project root (default: auto-detect)
+            auto_update: Whether to auto-rebuild on source changes
+        """
+        ...
+
+    @overload
+    def include(self, module_name: str = "fast_list", auto_update: Optional[bool] = None) -> Fast_listModuleWrapper: ...
+
+    @overload
+    def include(self, module_name: str = "game_math", auto_update: Optional[bool] = None) -> Game_mathModuleWrapper: ...
+
+    @overload
+    def include(self, module_name: str = "solar_system", auto_update: Optional[bool] = None) -> Solar_systemModuleWrapper: ...
+
+    @overload
+    def include(self, module_name: str = "templates", auto_update: Optional[bool] = None) -> TemplatesModuleWrapper: ...
+
+    @overload
+    def include(self, module_name: str, auto_update: Optional[bool] = None) -> Any: ...
+
+    def include(self, module_name: str, auto_update: Optional[bool] = None) -> Any:
+        """Load a C++ module.
+
+        Args:
+            module_name: Name of the module to load
+            auto_update: Override auto-update setting for this module
+
+        Returns:
+            ModuleWrapper with access to C++ classes, functions, and structs
+        """
+        ...
+
+    def rebuild(self, verbose: bool = False) -> bool:
+        """Rebuild all C++ modules."""
+        ...
+
+    def list_modules(self) -> List[str]:
+        """List available modules."""
+        ...