Revert "chore(core): make ty the default typechecker (#736)"

This reverts commit 052545b.

Author: phernandez

justfile

@@ -170,17 +170,13 @@ lint: fix
 fix:
     uv run ruff check --fix --unsafe-fixes src tests test-int
 
-# Type check code (ty)
-typecheck:
-    uv run ty check src tests test-int
-
 # Type check code (pyright)
-typecheck-pyright:
+typecheck:
     uv run pyright
 
 # Type check code (ty)
 typecheck-ty:
-    just typecheck
+    uv run ty check src/
 
 # Clean build artifacts and cache files
 clean:

src/basic_memory/api/v2/routers/prompt_router.py

@@ -6,7 +6,6 @@
 """
 
 from datetime import datetime, timezone
-from typing import Any
 from fastapi import APIRouter, HTTPException, status, Path
 from loguru import logger
 
@@ -60,7 +59,6 @@ async def continue_conversation(
 
     # Initialize search results
     search_results = []
-    hierarchical_results_for_count = []
 
     # Get data needed for template
     if request.topic:
@@ -93,8 +91,7 @@ async def continue_conversation(
         # Limit to a reasonable number of total results
         all_hierarchical_results = all_hierarchical_results[:10]
 
-        hierarchical_results_for_count = all_hierarchical_results
-        template_context: dict[str, Any] = {
+        template_context = {
             "topic": request.topic,
             "timeframe": request.timeframe,
             "hierarchical_results": all_hierarchical_results,
@@ -113,7 +110,6 @@ async def continue_conversation(
 
         hierarchical_results = recent_context.results[:5]  # Limit to top 5 recent items
 
-        hierarchical_results_for_count = hierarchical_results
         template_context = {
             "topic": f"Recent Activity from ({request.timeframe})",
             "timeframe": request.timeframe,
@@ -133,6 +129,9 @@ async def continue_conversation(
         relation_count = 0
         entity_count = 0
 
+        # Get the hierarchical results from the template context
+        hierarchical_results_for_count = template_context.get("hierarchical_results", [])
+
         # For topic-based search
         if request.topic:
             for item in hierarchical_results_for_count:
@@ -160,24 +159,29 @@ async def continue_conversation(
                             elif related.type == "entity":  # pragma: no cover
                                 entity_count += 1  # pragma: no cover
 
-        prompt_metadata = PromptMetadata(
-            query=request.topic,
-            timeframe=request.timeframe,
-            search_count=len(search_results) if request.topic else 0,
-            context_count=len(hierarchical_results_for_count),
-            observation_count=observation_count,
-            relation_count=relation_count,
-            total_items=(
+        # Build metadata
+        metadata = {
+            "query": request.topic,
+            "timeframe": request.timeframe,
+            "search_count": len(search_results)
+            if request.topic
+            else 0,  # Original search results count
+            "context_count": len(hierarchical_results_for_count),
+            "observation_count": observation_count,
+            "relation_count": relation_count,
+            "total_items": (
                 len(hierarchical_results_for_count)
                 + observation_count
                 + relation_count
                 + entity_count
             ),
-            search_limit=request.search_items_limit,
-            context_depth=request.depth,
-            related_limit=request.related_items_limit,
-            generated_at=datetime.now(timezone.utc).isoformat(),
-        )
+            "search_limit": request.search_items_limit,
+            "context_depth": request.depth,
+            "related_limit": request.related_items_limit,
+            "generated_at": datetime.now(timezone.utc).isoformat(),
+        }
+
+        prompt_metadata = PromptMetadata(**metadata)
 
         return PromptResponse(
             prompt=rendered_prompt, context=template_context, metadata=prompt_metadata
@@ -225,7 +229,7 @@ async def search_prompt(
     results = await search_service.search(query, limit=limit, offset=offset)
     search_results = await to_search_results(entity_service, results)
 
-    template_context: dict[str, Any] = {
+    template_context = {
         "query": request.query,
         "timeframe": request.timeframe,
         "results": search_results,
@@ -237,19 +241,22 @@ async def search_prompt(
         # Render template
         rendered_prompt = await template_loader.render("prompts/search.hbs", template_context)
 
-        prompt_metadata = PromptMetadata(
-            query=request.query,
-            timeframe=request.timeframe,
-            search_count=len(search_results),
-            context_count=len(search_results),
-            observation_count=0,
-            relation_count=0,
-            total_items=len(search_results),
-            search_limit=limit,
-            context_depth=0,
-            related_limit=0,
-            generated_at=datetime.now(timezone.utc).isoformat(),
-        )
+        # Build metadata
+        metadata = {
+            "query": request.query,
+            "timeframe": request.timeframe,
+            "search_count": len(search_results),
+            "context_count": len(search_results),
+            "observation_count": 0,  # Search results don't include observations
+            "relation_count": 0,  # Search results don't include relations
+            "total_items": len(search_results),
+            "search_limit": limit,
+            "context_depth": 0,  # No context depth for basic search
+            "related_limit": 0,  # No related items for basic search
+            "generated_at": datetime.now(timezone.utc).isoformat(),
+        }
+
+        prompt_metadata = PromptMetadata(**metadata)
 
         return PromptResponse(
             prompt=rendered_prompt, context=template_context, metadata=prompt_metadata

src/basic_memory/api/v2/routers/resource_router.py

@@ -211,7 +211,7 @@ async def create_resource(
                 action="create",
                 phase="search_index",
             ):
-                await search_service.index_entity(entity)
+                await search_service.index_entity(entity)  # pyright: ignore
 
             return ResourceResponse(
                 entity_id=entity.id,
@@ -326,16 +326,14 @@ async def update_resource(
                         "updated_at": file_metadata.modified_at,
                     },
                 )
-            if updated_entity is None:
-                raise HTTPException(status_code=404, detail=f"Entity {entity_id} not found")
 
             with telemetry.scope(
                 "api.resource.update.search_index",
                 domain="resource",
                 action="update",
                 phase="search_index",
             ):
-                await search_service.index_entity(updated_entity)
+                await search_service.index_entity(updated_entity)  # pyright: ignore
 
             return ResourceResponse(
                 entity_id=entity.id,