Add tests

.coveragerc

@@ -0,0 +1,7 @@
+[run]
+source = privatellm
+
+[report]
+exclude_lines =
+    pragma: no cover
+    @abstractmethod

.gitignore

@@ -3,4 +3,5 @@ files
 apikey.txt
 **/__pycache__
 db
-
+.coverage
+.vscode/

.pre-commit-config.yaml

@@ -0,0 +1,28 @@
+fail_fast: true
+repos:
+- repo: local
+  hooks:
+  - id: format-check
+    name: make format-check
+    entry: make format-check
+    language: system
+    pass_filenames: false
+    always_run: true
+  - id: lint
+    name: make lint
+    entry: make lint
+    language: system
+    pass_filenames: false
+    always_run: true
+  - id: mypy
+    name: make mypy
+    entry: make mypy
+    language: system
+    pass_filenames: false
+    always_run: true
+  - id: test
+    name: make test
+    entry: make test
+    language: system
+    pass_filenames: false
+    always_run: true

Makefile

@@ -0,0 +1,38 @@
+init:
+	poetry install
+	pre-commit install
+
+format:
+	poetry run ruff format .
+	poetry run ruff . --fix
+
+format-check:
+	poetry run ruff format . --check
+
+lint:
+	poetry run ruff .
+
+mypy:
+	poetry run mypy --install-types --non-interactive .
+
+
+test:
+	poetry run pytest -vv -s --typeguard-packages=privatellm
+
+coverage:
+	poetry run coverage run -m pytest
+	poetry run coverage html
+	poetry run coverage report
+
+checks: format-check lint mypy test
+
+image:
+	DOCKER_BUILDKIT=1 docker build --progress=plain -t texttitan:latest .
+
+docs:
+	mkdocs serve
+
+update:
+	poetry lock
+
+.PHONY: init format format-check lint mypy test coverage checks image docs update

README.md

@@ -11,8 +11,9 @@ hacked during codecamp 2023.
 # install dependencies
 poetry install
 
-# start postgres vectorstore
+# start postgres vectorstore and jaeger trace collector
 docker compose up -d
+export OTEL_EXPORTER_OTLP_ENDPOINT="http://localhost:4317"
 
 # run texttitan
 poetry run python privatellm/main.py

mypy.ini

@@ -0,0 +1,26 @@
+[mypy]
+disallow_untyped_calls = True
+disallow_untyped_defs = True
+disallow_incomplete_defs = True
+disallow_untyped_decorators = True
+disallow_any_generics = True
+disallow_subclassing_any = True
+
+warn_return_any = True
+warn_redundant_casts = True
+warn_unused_ignores = True
+warn_unused_configs = True
+
+ignore_missing_imports = True
+follow_imports = silent
+exclude = (?x)(
+    build/lib/*/*
+    | tests/*
+    | venv/Scripts/*
+    | venv/bin/*
+  )
+
+
+
+[mypy-tests.*]
+ignore_errors = True