memos-engine 0.1.0__tar.gz → 0.2.0__tar.gz

memos_engine-0.2.0/.github/release.yml

@@ -0,0 +1,10 @@
+changelog:
+  categories:
+    - title: Features
+      labels: ["feat"]
+    - title: Bug Fixes
+      labels: ["fix"]
+    - title: Documentation
+      labels: ["docs"]
+    - title: Other Changes
+      labels: ["*"]

{memos_engine-0.1.0 → memos_engine-0.2.0}/.github/workflows/release.yml

@@ -19,3 +19,7 @@ jobs:
       - run: uv publish
         env:
           UV_PUBLISH_TOKEN: ${{ secrets.PYPI_TOKEN }}
+      - uses: softprops/action-gh-release@v2
+        with:
+          files: dist/*
+          generate_release_notes: true

{memos_engine-0.1.0 → memos_engine-0.2.0}/.gitignore

@@ -3,6 +3,9 @@ __pycache__/
 *.py[codz]
 *$py.class
 
+# memos local index database
+.memos/
+
 MEMORY_OS_ARCHITECTURE.md
 
 # C extensions

memos_engine-0.2.0/AGENTS.md

@@ -0,0 +1,263 @@
+# AGENTS.md
+
+> Этот репозиторий проиндексирован `memos` — используй MCP-инструменты
+> memos для навигации по коду. Это быстрее grep/cat и точнее.
+
+## Code navigation policy (memos)
+
+Этот проект проиндексирован `memos` — структурным индексом кода (symbols,
+call edges, imports) с episodic memory. Инструмент доступен через MCP-тулы.
+**Прежде чем читать файл целиком или делать grep по коду — используй memos.**
+Это быстрее, точнее и не тратит контекст на нерелевантные строки.
+
+### Обязательный порядок действий
+
+1. **В начале сессии**, если ещё не открыт проект — вызови `open_project`
+   с абсолютным путём к корню репозитория. Это одноразовое действие на
+   сессию, если проект уже открыт — не вызывай повторно.
+
+2. **Чтобы найти определение функции/класса/переменной** — используй
+   `find_symbol_tool(name, kind=None, file_path=None)`.
+   Не используй `grep`/`cat` для этого — memos даёт точный результат
+   (файл, строки, сигнатура, exported) без ложных совпадений по имени
+   в комментариях/строках.
+
+3. **Чтобы понять, кто вызывает функцию или что вызывает она** — используй
+   `find_calls_tool(symbol_name, direction="callers"|"callees")`.
+   Это заменяет ручной grep по имени функции во всём репозитории.
+
+4. **Перед тем как редактировать функцию** — вызови `get_context_tool(symbol_id)`.
+   Он возвращает символ + всех callers + callees + episodic memory заметки +
+   кэшированный LLM-summary (если есть). **Это заменяет повторное чтение
+   всего файла и всех зависимых файлов** — не открывай их вручную, если
+   `get_context_tool` уже дал нужную картину.
+
+5. **Чтобы посмотреть всё содержимое файла структурно** (какие символы,
+   вызовы, импорты) — используй `get_module_tool(path)` вместо чтения
+   файла целиком, если тебе не нужен именно raw-текст (форматирование,
+   комментарии, точный синтаксис для правки).
+
+6. **Для поиска по смыслу, а не по точному имени** ("где у нас логика
+   авторизации", "функция, которая парсит даты") — используй
+   `semantic_search_tool(query, top_k)`.
+
+7. **После значимого решения, вывода или "гочи"**, которое стоит помнить
+   в будущих сессиях (архитектурное решение, причина странного кода,
+   TODO с контекстом) — сохрани заметку через `memory_add_note(content,
+   scope_type, scope_id, kind)`. Не полагайся только на свой ответ в чате —
+   он не переживёт сессию, а memory_entries переживают.
+
+8. **Перед началом работы с заметками** — вызови `get_memories(scope_type,
+   scope_id)`, чтобы узнать, что уже известно про этот файл/символ/проект
+   из прошлых сессий, прежде чем исследовать заново.
+
+### Когда grep/cat/read_file всё-таки уместны
+
+- Файл ещё не проиндексирован (новый, не .ts/.tsx/.go — расширения см. ниже)
+  или недавно создан и явно устарел в индексе
+- Нужен именно raw-текст: точное форматирование, конкретные комментарии,
+  импорты в исходном виде для правки строки
+- `find_symbol_tool`/`semantic_search_tool` не нашли ничего релевантного —
+  тогда grep как fallback, но перепроверь опечатки в имени символа сначала
+
+### Что НЕ делать
+
+- Не грепай по имени функции, чтобы найти её вызовы — используй
+  `find_calls_tool`. Grep даёт ложные совпадения (комментарии, строки,
+  похожие имена в других scope) и не различает caller/callee направление.
+- Не читай весь файл, чтобы понять один символ — `get_context_tool` уже
+  дал тебе минимально достаточный контекст.
+- Не забывай `open_project`, если сессия началась с чистого MCP-сервера —
+  без этого все остальные тулы вернут ошибку "no project open".
+- Не дублируй информацию, которая уже есть в `get_memories` — сначала
+  проверь память, потом исследуй заново.
+
+### Available tools cheat sheet
+
+| Tool | Когда использовать |
+|------|---------------------|
+| `open_project(path)` | В начале сессии, один раз на проект |
+| `find_symbol_tool(name, kind?, file_path?)` | Найти определение символа по имени |
+| `find_calls_tool(symbol_name, direction)` | Callers или callees символа |
+| `get_module_tool(path)` | Всё содержимое файла структурно |
+| `get_context_tool(symbol_id)` | Полный контекст перед правкой функции |
+| `rename_impact_tool(symbol_id)` | Анализ последствий переименования символа |
+| `diff_impact_tool(path)` | Анализ blast radius exported-символов файла |
+| `semantic_search_tool(query, top_k?)` | Поиск по смыслу, не по точному имени |
+| `list_files_tool(path_filter?)` | Список индексированных файлов |
+| `list_symbols_tool(file_path?, kind?)` | Список индексированных символов |
+| `list_projects_tool()` | Инфо и статистика по текущему проекту |
+| `memory_add_note(content, scope_type, scope_id?, kind?)` | Сохранить заметку/решение |
+| `get_memories(scope_type?, scope_id?)` | Прочитать заметки из прошлых сессий |
+| `find_unused_symbols_tool()` | Найти неиспользуемые private функции |
+| `find_dead_imports_tool()` | Найти неразрешившиеся импорты |
+| `get_dependency_graph_tool()` | Граф зависимостей между файлами |
+| `find_import_cycles_tool()` | Найти циклы в импортах |
+| `memory_search_tool(query, top_k?)` | Полнотекстовый поиск по memory entries |
+| `memory_prune_tool(older_than_days?, kind?, apply?)` | Удаление устаревших записей (dry-run по умолчанию) |
+| `reindex_file_tool(path)` | Переиндексировать один файл после правки |
+
+Индексируются файлы с расширениями `.ts`, `.tsx`, `.go`. Индекс лежит в
+`.memos/memory.db` в корне проекта и обновляется через `memos index --path .`
+(инкрементально, по content hash — быстро на повторных запусках). Для
+переиндексации одного файла после правки используй `reindex_file_tool(path)`
+— это быстрее и дешевле полного прохода.
+
+Если каких-то тулов из таблицы нет в твоём MCP-клиенте — значит memos ещё
+не переиндексирован для этого проекта или сервер не запущен: запусти
+`memos serve-mcp` и настрой подключение согласно README проекта memos.
+
+
+## Workflow
+
+After completing each task: `ruff check .` + `pytest` green, then commit with
+a short descriptive title and all details in the commit body.
+
+## Commands
+
+| Command | What |
+|---------|------|
+| `uv run pytest` | all tests |
+| `uv run pytest -v` | verbose |
+| `uv run pytest tests/test_crud.py::test_symbol_crud` | single test |
+| `uv run memos index --path . --full` | index project at path |
+| `uv run memos query symbol <name> [--kind KIND]` | find symbols by name |
+| `uv run memos query calls <name> [--direction callers\|callees]` | find callers/callees |
+| `uv run memos query module <path>` | show everything for a file |
+| `uv run memos serve --path . --port 8000` | start FastAPI server |
+| `uv run memos serve-mcp --path .` | start MCP server (stdio) |
+| `uv run pytest tests/test_mcp.py` | MCP server tests |
+| `uv run pytest tests/test_llm_summary.py` | LLM summary tests |
+| `uv run pytest tests/test_impact.py` | Impact analysis tests |
+| `uv run pytest tests/test_hygiene.py` | Dead code / hygiene tests |
+| `uv run pytest tests/test_dependency_graph.py` | Dependency graph tests |
+| `uv run pytest tests/test_memory_hygiene.py` | Memory search & prune tests |
+| `uv run pytest tests/test_reindex.py` | Reindex tool tests |
+| `curl http://localhost:8000/symbols/{id}/context` | API: get context for symbol |
+| `uv add <pkg>` | add dependency |
+| `uv add --dev <pkg>` | add dev dependency |
+| `uv sync` | reinstall after pyproject.toml changes |
+
+## Layout
+
+```
+memos/
+  core/
+    db.py           # get_connection(WAL+FK), run_migrations, CRUD (all return pydantic models), resolve_call_edges
+    models.py       # Project, File, Symbol, CallEdge, Import, MemoryEntry
+    schema.sql      # human-readable DDL copy
+    migrations/
+      0001_init.sql # authoritative DDL
+      0002_vec.sql
+      0003_prompt_version.sql
+      0004_memory_fts.sql
+  indexer/
+    base.py         # LanguageIndexer ABC + ParsedSymbol/Call/Import/Result dataclasses
+    typescript.py   # TypeScriptIndexer (tree-sitter, handles .ts + .tsx)
+    go.py           # GoIndexer (tree-sitter, export by name case)
+    python.py       # PythonIndexer (tree-sitter, export by _ prefix)  # NEW
+    diff.py         # compute_file_hash, should_reindex
+  query/
+    core.py         # find_symbol, find_calls, get_module, find_calls_by_id, semantic_search,
+                    # list_files, list_symbols, get_or_generate_summary, get_context,
+                    # get_rename_impact, get_diff_impact, find_unused_symbols, find_dead_imports,
+                    # get_dependency_graph, find_import_cycles, memory_search, memory_prune
+                    # — pure query layer over db
+  api/
+    main.py         # FastAPI app — thin adapter over query/core.py
+    schemas.py      # pydantic response models for API
+  mcp/
+    server.py       # MCP server (FastMCP, stdio) — thin adapter over query/core.py
+  search/
+    base.py         # EmbeddingModel + VectorStore ABCs
+    embeddings.py   # FastEmbedEmbedding (all-MiniLM-L6-v2, 384-dim)
+    sqlite_vec_store.py  # SqliteVecStore (sqlite-vec vec0 table)
+  cli/
+    main.py         # argparse: "memos index [--path .] [--full] [--no-embed]"
+                    #          "memos query (symbol|calls|module)"
+                    #          "memos serve [--path] [--port]"
+                    #          "memos serve-mcp [--path]"
+tests/
+  conftest.py       # fixture: in-memory sqlite with migrations applied
+  test_schema.py    # table existence checks
+  test_crud.py      # CRUD + cascade delete
+  test_migrations.py# idempotent re-run
+   test_typescript_indexer.py  # 18 unit tests on TS parsing
+   test_go_indexer.py          # 18 unit tests on Go parsing
+   test_python_indexer.py      # 18 unit tests on Python parsing
+   test_cli_index.py           # 11 integration tests: index flow (TS + Go + Python)
+   test_resolver.py            # 8 unit tests on call-edge resolution (+ Python)
+   test_query.py               # 12 unit tests on query/core.py
+   test_cli_query.py           # 10 integration tests: query flow (TS + Go + Python)
+  test_api.py                 # 7 integration tests: FastAPI endpoints
+  test_mcp.py                 # 15 tests: MCP server tools
+  test_semantic_search.py     # 8 tests: VecStore CRUD + semantic_search query
+  test_llm_summary.py         # 11 tests: get_or_generate_summary, get_context, source_hash
+  test_impact.py              # 8 tests: rename_impact, diff_impact
+  test_hygiene.py             # 8 tests: unused symbols, dead imports
+  test_dependency_graph.py    # 8 tests: dependency graph + cycle detection
+  test_memory_hygiene.py      # 9 tests: memory search + prune
+  test_reindex.py             # 3 tests: reindex_file_tool
+  test_migrations.py          # 2 tests: idempotent re-run
+  fixtures/
+    typescript_mini/src/    # 3 .ts files for integration testing
+    go_mini/src/            # 3 .go files for integration testing
+    python_mini/src/        # 3 .py files for integration testing  # NEW
+    import_cycle/src/       # 2 .ts files with circular import
+```
+
+## Dependencies
+
+- **pydantic** — all CRUD returns models, not raw rows
+- **tree-sitter + tree-sitter-typescript + tree-sitter-go + tree-sitter-python** — AST parsing (.ts, .tsx, .go, .py)
+- **stdlib sqlite3** — connection mgmt, WAL journal, FK enforcement
+- **fastapi + uvicorn** — HTTP API
+- **mcp[cli]** — MCP server (FastMCP, stdio transport)
+- **fastembed** — ONNX embeddings (all-MiniLM-L6-v2, 384-dim)
+- **sqlite-vec** — vector search extension for sqlite
+- **rich** — CLI progress bars
+- **pytest** (dev)
+- **httpx** (dev, for TestClient)
+- **pytest-anyio** (dev, for async MCP tests)
+- **hatchling** (build)
+
+## Architecture conventions
+
+- `indexer/typescript.py` produces plain dataclasses (`ParseResult`), not pydantic models — conversion happens in CLI when calling CRUD
+- `cli/main.py` is a thin argparse wrapper over `indexer/` + `core/db.py`; no business logic in CLI
+- `query/core.py` has no knowledge of CLI, API, or MCP — all three are thin adapters over it
+- `mcp/server.py` is a thin FastMCP wrapper over `query/core.py`; same pattern as `api/main.py`
+- All MCP tools return JSON strings (parsable by the LLM), never raw text
+- `index_file()` in cli/main.py is the unit of change: delete old → parse → insert new
+- `memos index` stores DB at `{project_root}/.memos/memory.db`
+- call_edges are inserted with `callee_symbol_id=NULL` (first pass); resolution is Task 5
+- `callee_symbol_id` and `resolved_file_id` use `ON DELETE SET NULL` — when a callee symbol is deleted (e.g. during reindex), the FK is automatically nulled, then re-resolved in the second pass
+- `export` keyword detection (TS): `export_statement` wraps declarations; the walker passes `exported=True` to children
+- Go export: determined by `name[0].isupper()` — no `export_statement`
+- Python export: determined by `name[0] != '_'` — convention-based, like Go
+- Go methods: `method_declaration` nodes are separate from type declarations; receiver type is extracted from `receiver` field
+- Go imports: both single `import "x"` and grouped `import ( "x" "y" )` forms are handled via `import_spec` / `import_spec_list`
+- Python imports: `import_statement` (single/as) and `import_from_statement` (from/relative) — no resolver yet
+
+## What does not exist yet
+
+- Type checking, codegen — none configured
+
+## Execution order (from spec §6)
+
+1. ✅ `core/db.py` + `models.py` + `schema.sql` + migrations + tests
+2. ✅ `indexer/base.py` + `indexer/typescript.py` + CLI `index`
+3. ✅ `indexer/go.py`
+4. ✅ `query/core.py` (find_symbol, find_calls, get_module) + CLI `query`
+5. ✅ Call-edge resolution (second pass) + cross-file tests
+6. ✅ FastAPI wrapper
+7. ✅ Semantic search (sqlite-vec + sentence-transformers)
+8. ✅ MCP server
+9. ✅ `memory_entries` write-path via `memory_add_note`
+10. ✅ LLM enrichment: `get_or_generate_summary()` with content_hash check, `get_context()` composite function
+11. ✅ Section 0: Technical debt (dedup, batch resolve, SAVEPOINT, threading.Lock)
+12. ✅ Section 1: Impact analysis (rename_impact, diff_impact, MCP, API, tests)
+13. ✅ Section 2: Dead code / hygiene (unused symbols, dead imports, MCP, API, tests)
+14. ✅ Section 3: Dependency graph (graph + cycle detection, MCP, API, tests)
+15. ✅ Section 4: Memory search & hygiene (FTS5 search, prune, MCP, API, tests)
+16. ✅ Section 5: Reindex file tool (per-file reindex via MCP, tests)

memos_engine-0.2.0/CHANGELOG.md

@@ -0,0 +1,19 @@
+# Changelog
+
+## 0.2.0 — 2026-07-21
+
+- Python language support (indexer, tests, fixtures)
+- Self-indexed: memos-engine uses memos for its own development
+- PyPI package `memos-engine` with MIT license
+- CI: GitHub Actions release workflow (tag → build → publish)
+- GitHub Release automation with changelog categorization
+
+## 0.1.0 — 2026-07-21
+
+- Initial release
+- TypeScript/Go structural index via tree-sitter
+- MCP server with 20 tools
+- FastAPI with 15 endpoints
+- Semantic search (fastembed + sqlite-vec)
+- Episodic memory (FTS5)
+- Impact analysis, dead code, dependency graph, memory hygiene

{memos_engine-0.1.0 → memos_engine-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: memos-engine
-Version: 0.1.0
+Version: 0.2.0
 Summary: Memory OS for AI coding agents — structural code index with episodic memory
 Project-URL: Repository, https://github.com/TAskMAster339/memos-engine
 Author: TAskMAster339
@@ -18,6 +18,7 @@ Requires-Dist: pydantic
 Requires-Dist: rich>=15.0.0
 Requires-Dist: sqlite-vec>=0.1.0
 Requires-Dist: tree-sitter-go>=0.25.0
+Requires-Dist: tree-sitter-python>=0.23.0
 Requires-Dist: tree-sitter-typescript>=0.23.2
 Requires-Dist: tree-sitter>=0.26.0
 Requires-Dist: uvicorn[standard]>=0.34.0
@@ -27,10 +28,20 @@ Description-Content-Type: text/markdown
 
 [GitHub: TAskMAster339/memos-engine](https://github.com/TAskMAster339/memos-engine.git)
 
+<div align="center">
+
+![Python](https://img.shields.io/badge/Python-3776AB?logo=python&logoColor=fff)
+![TypeScript](https://img.shields.io/badge/TypeScript-3178C6?logo=typescript&logoColor=fff)
+![TSX](https://img.shields.io/badge/TSX-3178C6?logo=react&logoColor=fff)
+![Go](https://img.shields.io/badge/Go-00ADD8?logo=go&logoColor=fff)
+
+</div>
+
 `memos` builds a **structural index** (symbols, call edges, imports) of a
-TypeScript/TSX / Go codebase using tree-sitter and stores it in SQLite. It is the
-first layer of a larger *Memory OS* for AI coding agents — instead of
-grepping text, agents query **structure** (definitions, callers, callees).
+TypeScript / TSX / Go / Python codebase using tree-sitter and stores it in
+SQLite. It is the first layer of a larger *Memory OS* for AI coding agents
+— instead of grepping text, agents query **structure** (definitions,
+callers, callees).
 
 ## Integrating memos into your project
 

{memos_engine-0.1.0 → memos_engine-0.2.0}/README.md

@@ -2,10 +2,20 @@
 
 [GitHub: TAskMAster339/memos-engine](https://github.com/TAskMAster339/memos-engine.git)
 
+<div align="center">
+
+![Python](https://img.shields.io/badge/Python-3776AB?logo=python&logoColor=fff)
+![TypeScript](https://img.shields.io/badge/TypeScript-3178C6?logo=typescript&logoColor=fff)
+![TSX](https://img.shields.io/badge/TSX-3178C6?logo=react&logoColor=fff)
+![Go](https://img.shields.io/badge/Go-00ADD8?logo=go&logoColor=fff)
+
+</div>
+
 `memos` builds a **structural index** (symbols, call edges, imports) of a
-TypeScript/TSX / Go codebase using tree-sitter and stores it in SQLite. It is the
-first layer of a larger *Memory OS* for AI coding agents — instead of
-grepping text, agents query **structure** (definitions, callers, callees).
+TypeScript / TSX / Go / Python codebase using tree-sitter and stores it in
+SQLite. It is the first layer of a larger *Memory OS* for AI coding agents
+— instead of grepping text, agents query **structure** (definitions,
+callers, callees).
 
 ## Integrating memos into your project
 

{memos_engine-0.1.0 → memos_engine-0.2.0}/memos/cli/main.py

@@ -32,6 +32,7 @@ from memos.core.db import (
 from memos.core.models import CallEdge, File, Import, Project, Symbol
 from memos.indexer.diff import compute_file_hash, should_reindex
 from memos.indexer.go import GoIndexer
+from memos.indexer.python import PythonIndexer
 from memos.indexer.typescript import TypeScriptIndexer
 from memos.query.core import (
     add_memory_entry,
@@ -46,6 +47,7 @@ EXTENSION_INDEXERS = {
     ".ts": TypeScriptIndexer(tsx=False),
     ".tsx": TypeScriptIndexer(tsx=True),
     ".go": GoIndexer(),
+    ".py": PythonIndexer(),
 }
 
 SKIP_DIRS = {".git", ".memos", "node_modules", "__pycache__", ".venv", "target"}

memos_engine-0.2.0/memos/indexer/python.py

@@ -0,0 +1,184 @@
+import hashlib
+
+import tree_sitter_python
+from tree_sitter import Language, Parser
+
+from memos.indexer.base import (
+    LanguageIndexer,
+    ParsedCall,
+    ParsedImport,
+    ParsedSymbol,
+    ParseResult,
+)
+
+_PY_LANG = Language(tree_sitter_python.language())
+
+
+class PythonIndexer(LanguageIndexer):
+    def __init__(self):
+        self.parser = Parser()
+        self.parser.language = _PY_LANG
+
+    def language(self) -> str:
+        return "python"
+
+    def parse(self, source: str, file_path: str) -> ParseResult:
+        source_bytes = bytes(source, "utf-8")
+        tree = self.parser.parse(source_bytes)
+        result = ParseResult()
+        self._walk(tree.root_node, source_bytes, result)
+        return result
+
+    def _walk(self, node, source, result, scope=None, class_name=None):  # noqa: C901, PLR0911, PLR0912
+        t = node.type
+
+        if t == "function_definition":
+            name_node = node.child_by_field_name("name")
+            name = self._node_text(name_node, source) if name_node else None
+            if name:
+                kind = "method" if class_name else "function"
+                self._add_symbol(
+                    node, source, result, kind, name, parent_name=class_name,
+                )
+                method_scope = f"{class_name}.{name}" if class_name else name
+                for child in node.children:
+                    self._walk(
+                        child, source, result,
+                        scope=method_scope, class_name=class_name,
+                    )
+            return
+
+        if t == "class_definition":
+            name_node = node.child_by_field_name("name")
+            name = self._node_text(name_node, source) if name_node else None
+            if name:
+                self._add_symbol(node, source, result, "class", name)
+                for child in node.children:
+                    self._walk(child, source, result, scope=name, class_name=name)
+            return
+
+        if t == "decorated_definition":
+            for child in node.children:
+                self._walk(child, source, result, scope=scope, class_name=class_name)
+            return
+
+        if t == "expression_statement":
+            if scope is None and class_name is None:
+                for child in node.children:
+                    if child.type == "assignment":
+                        self._handle_assignment(child, source, result)
+            for child in node.children:
+                self._walk(child, source, result, scope=scope, class_name=class_name)
+            return
+
+        if t == "call":
+            self._add_call(node, source, result, scope)
+            for child in node.children:
+                self._walk(child, source, result, scope=scope, class_name=class_name)
+            return
+
+        if t == "import_statement":
+            self._add_import(node, source, result)
+            return
+
+        if t == "import_from_statement":
+            self._add_from_import(node, source, result)
+            return
+
+        for child in node.children:
+            self._walk(child, source, result, scope=scope, class_name=class_name)
+
+    def _handle_assignment(self, node, source, result):
+        left = node.child_by_field_name("left")
+        if left and left.type == "identifier":
+            name = self._node_text(left, source)
+            kind = "const" if name.isupper() else "variable"
+            text = source[node.start_byte : node.end_byte].decode("utf-8")
+            content_hash = hashlib.sha256(text.encode()).hexdigest()
+            exported = not name.startswith("_")
+            result.symbols.append(
+                ParsedSymbol(
+                    name=name,
+                    kind=kind,
+                    signature=text,
+                    start_line=node.start_point[0] + 1,
+                    end_line=node.end_point[0] + 1,
+                    exported=exported,
+                    content_hash=content_hash,
+                ),
+            )
+
+    def _add_symbol(self, node, source, result, kind, name, parent_name=None):  # noqa: PLR0913
+        text = source[node.start_byte : node.end_byte].decode("utf-8")
+        content_hash = hashlib.sha256(text.encode()).hexdigest()
+        exported = not name.startswith("_")
+
+        if kind in ("function", "method"):
+            sig = self._extract_signature(node, source, text)
+        else:
+            sig = text.split("{", 1)[0].strip() if "{" in text else text
+
+        result.symbols.append(
+            ParsedSymbol(
+                name=name,
+                kind=kind,
+                signature=sig,
+                start_line=node.start_point[0] + 1,
+                end_line=node.end_point[0] + 1,
+                exported=exported,
+                content_hash=content_hash,
+                parent_name=parent_name,
+            ),
+        )
+
+    def _extract_signature(self, node, source, text):
+        params = node.child_by_field_name("parameters")
+        if params is None:
+            return None
+        params_end = params.end_byte - node.start_byte
+        return_type = node.child_by_field_name("return_type")
+        if return_type:
+            params_end = return_type.end_byte - node.start_byte
+        colon_idx = text.find(":", params_end)
+        if colon_idx != -1:
+            return text[: colon_idx + 1].strip()
+        return text
+
+    def _add_call(self, node, source, result, caller_name):
+        func_node = node.child_by_field_name("function")
+        if func_node is None:
+            func_node = node.children[0] if node.children else None
+        if func_node is None:
+            return
+        if func_node.type in ("identifier", "attribute"):
+            name = self._node_text(func_node, source)
+        else:
+            return
+        result.calls.append(
+            ParsedCall(
+                caller_name=caller_name,
+                callee_name=name,
+                line=node.start_point[0] + 1,
+            ),
+        )
+
+    def _add_import(self, node, source, result):
+        for child in node.children:
+            if child.type == "dotted_name":
+                path = self._node_text(child, source)
+                result.imports.append(ParsedImport(imported_path=path))
+            elif child.type == "aliased_import":
+                name_node = child.child_by_field_name("name")
+                if name_node:
+                    path = self._node_text(name_node, source)
+                    result.imports.append(ParsedImport(imported_path=path))
+
+    def _add_from_import(self, node, source, result):
+        module_node = node.child_by_field_name("module_name")
+        if module_node:
+            path = self._node_text(module_node, source)
+            result.imports.append(ParsedImport(imported_path=path))
+
+    @staticmethod
+    def _node_text(node, source):
+        return source[node.start_byte : node.end_byte].decode("utf-8")

{memos_engine-0.1.0 → memos_engine-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "memos-engine"
-version = "0.1.0"
+version = "0.2.0"
 description = "Memory OS for AI coding agents — structural code index with episodic memory"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -23,6 +23,7 @@ dependencies = [
     "sqlite-vec>=0.1.0",
     "rich>=15.0.0",
     "mcp[cli]>=1.28.1",
+    "tree-sitter-python>=0.23.0",
 ]
 
 [project.urls]

memos_engine-0.2.0/tests/fixtures/python_mini/src/main.py

@@ -0,0 +1,5 @@
+from .utils import greet
+
+
+def main():
+    greet("world")

memos_engine-0.2.0/tests/fixtures/python_mini/src/types.py

@@ -0,0 +1,11 @@
+class User:
+    name: str
+    age: int
+
+    def get_name(self) -> str:
+        return self.name
+
+
+class Printer:
+    def print(self, msg: str) -> None:
+        pass

memos_engine-0.2.0/tests/fixtures/python_mini/src/utils.py

@@ -0,0 +1,9 @@
+def _helper():
+    return 1
+
+
+def greet(name):
+    return f"hello {name}"
+
+
+PUBLIC_CONST = 42