Initialize with basic reviewer

.gitignore

@@ -0,0 +1,174 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# UV
+#   Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#uv.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/latest/usage/project/#working-with-version-control
+.pdm.toml
+.pdm-python
+.pdm-build/
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+
+# Ruff stuff:
+.ruff_cache/
+
+# PyPI configuration file
+.pypirc

.python-version

@@ -0,0 +1 @@
+3.11

README.md

@@ -0,0 +1,4 @@
+# LLM Review
+
+An agent that uses an LLM (OpenAI GPT or Anthropic Claude) to review a board
+schematic.

main.py

@@ -0,0 +1,131 @@
+import argparse
+import asyncio
+import json
+import logging
+import time
+
+from dataclasses import dataclass
+from typing import Optional
+
+from pydantic_ai import Agent, BinaryContent, RunContext
+
+
+VERSION = (0, 1, 0)
+
+SYSTEM_PROMPT = """
+You are a Schematic Engineer. You have been tasked with reviewing a schematic.
+The schematic will be provided to you as an image of the schematic. You have
+access to a tool which can provide you with the details of a component given
+its designator. You must review the schematic and provide a markdown document
+with the following details:
+
+1. Potential issues with the schematic.
+2. Any missing components.
+3. Scope for improvement in the schematic.
+
+Your suggestions must NOT require a human to further review. For example,
+instead of saying "Verify if all ICs have adequate decoupling capacitors", you
+should do the verification yourself and provide the results. All of your
+recommendations MUST include a reference to WHAT component, pin or net should be
+reviewed.
+
+You MUST use the tool provided to get the details of the components. The tool
+will provide you with the details of a component given its designator. DO NOT
+use the tool to fetch component designators that are NOT in the schematic image.
+
+If any of the details are not applicable, please mention so in your response.
+"""
+
+logger = logging.getLogger(__name__)
+handler = logging.StreamHandler()
+formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+handler.setFormatter(formatter)
+logger.addHandler(handler)
+
+# Global semaphore and timestamp for rate limiting
+component_call_limiter = asyncio.Semaphore(1)
+last_component_call_time = 0
+
+
+@dataclass
+class AgentDeps:
+    json: dict
+
+
+agent = Agent(
+    "openai:gpt-4o-mini",
+    deps_type=AgentDeps,
+    system_prompt=SYSTEM_PROMPT,
+)
+
+
+@agent.tool()
+async def get_component_details(
+    ctx: RunContext[AgentDeps], designator: str
+) -> Optional[dict]:
+    global last_component_call_time
+
+    async with component_call_limiter:
+        current_time = time.time()
+        time_since_last_call = current_time - last_component_call_time
+
+        # Rate limit: one call per 30 seconds
+        if time_since_last_call < 30:
+            wait_time = 30 - time_since_last_call
+            logger.info(
+                f"Rate limiting: waiting {wait_time:.2f} seconds before next component lookup"
+            )
+            await asyncio.sleep(wait_time)
+
+        # Update the last call time after potentially waiting
+        last_component_call_time = time.time()
+
+        logger.debug("Getting details for component %s", designator)
+        json = ctx.deps.json
+        first_page: dict = next(iter(json.get("pages", [])), {})
+        components: dict = first_page.get("components", {})
+        component = next(
+            (c for c in components.values() if c.get("reference") == designator), None
+        )
+
+        if component is None:
+            logger.warning("Component %s not found", designator)
+
+        return component
+
+
+async def main():
+    arg_parser = argparse.ArgumentParser(description="LLM Review")
+    arg_parser.add_argument(
+        "--json-path", help="Path to the JSON of the schematic", required=True
+    )
+    arg_parser.add_argument(
+        "--image-path", help="Path to a JPEG image of the schematic", required=True
+    )
+    arg_parser.add_argument(
+        "--output-path", help="Path to the output markdown", required=True
+    )
+    arg_parser.add_argument("--log-level", help="Log level", default="INFO")
+
+    args = arg_parser.parse_args()
+    logger.setLevel(args.log_level)
+
+    logger.info("LLM Review version %s" % ".".join(map(str, VERSION)))
+
+    with open(args.json_path, "r") as f:
+        json_data = json.load(f)
+    with open(args.image_path, "rb") as f:
+        image = f.read()
+
+    deps = AgentDeps(json=json_data)
+    result = await agent.run(
+        ["Review this schematic.", BinaryContent(data=image, media_type="image/jpeg")],
+        deps=deps,
+    )
+
+    with open(args.output_path, "w") as f:
+        f.write(result.data)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

pyproject.toml

@@ -0,0 +1,9 @@
+[project]
+name = "llm-review"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = [
+    "pydantic-ai>=0.0.41",
+]