typeagent-py 0.1.0__tar.gz

typeagent_py-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+    MIT License
+
+    Copyright (c) Microsoft Corporation.
+
+    Permission is hereby granted, free of charge, to any person obtaining a copy
+    of this software and associated documentation files (the "Software"), to deal
+    in the Software without restriction, including without limitation the rights
+    to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+    copies of the Software, and to permit persons to whom the Software is
+    furnished to do so, subject to the following conditions:
+
+    The above copyright notice and this permission notice shall be included in all
+    copies or substantial portions of the Software.
+
+    THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+    IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+    FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+    AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+    LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+    OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+    SOFTWARE

typeagent_py-0.1.0/PKG-INFO

@@ -0,0 +1,28 @@
+Metadata-Version: 2.4
+Name: typeagent-py
+Version: 0.1.0
+Summary: TypeAgent implements an agentic memory framework.
+Author: Steven Lucco, Umesh Madan, Guido van Rossum
+Author-email: Guido van Rossum <gvanrossum@microsoft.com>
+Project-URL: Homepage, https://github.com/microsoft/TypeAgent/tree/main/python/ta
+Requires-Python: <3.14,>=3.12
+License-File: LICENSE
+Requires-Dist: azure-identity>=1.22.0
+Requires-Dist: black>=25.1.0
+Requires-Dist: build>=1.2.2.post1
+Requires-Dist: colorama>=0.4.6
+Requires-Dist: coverage[toml]>=7.9.1
+Requires-Dist: logfire>=4.1.0
+Requires-Dist: mcp[cli]>=1.12.1
+Requires-Dist: numpy>=2.2.6
+Requires-Dist: openai>=1.81.0
+Requires-Dist: opentelemetry-instrumentation-httpx>=0.57b0
+Requires-Dist: pydantic>=2.11.4
+Requires-Dist: pydantic-ai-slim[openai]>=0.5.0
+Requires-Dist: pyright>=1.1.405
+Requires-Dist: pytest>=8.3.5
+Requires-Dist: pytest-asyncio>=0.26.0
+Requires-Dist: pytest-mock>=3.14.0
+Requires-Dist: python-dotenv>=1.1.0
+Requires-Dist: typechat
+Dynamic: license-file

typeagent_py-0.1.0/README.md

@@ -0,0 +1,60 @@
+# Python package 'typeagent'
+
+### This is an experimental prototype
+
+Working toward a shared understanding of the MVP for structured RAG.
+
+### This is sample code
+
+This is an in-progress project aiming at a Pythonic translation of
+`TypeAgent/ts/packages/knowPro` and a few related packages to Python.
+(Pythonic because it uses Python conventions and types as appropriate.)
+
+### Original description
+
+(Not all of this matches the current state of the code.)
+
+- Python class names correspond 1:1 to TS interface or type names.
+  (Much has changed here.)
+- Field and method names are converted from camelCase to python_case.
+- I-named interfaces become `Protocol` classes.
+  (Or in some cases `ABC`s.)
+- Other interfaces and structured types become dataclasses.
+- Union types remain union types.
+- Except unions of string literals become `Literal` types.
+- Not all of the TS class hierarchy is translated literally.
+
+### How to build
+
+Tested on Mac and on Ubuntu 22 under WSL.
+Should work on most UNIXoids.
+(For Windows, look at the `make.bat` script.)
+
+- Install Python 3.12 or higher (get it from
+  [python.org](https://www.python.org/downloads/) for Mac,
+  or run `sudo apt install python3.12` for Linux)
+- Run `make all`
+- You should now have a wheel file under `dist/`
+- To install: `python3.12 -m pip install dist/typeagent-0.1.0-py3-none-any.whl`
+- TODO: Upload that wheel to PyPI
+- To clean up, run `make clean`
+
+### How to test
+
+- Set your environment to contain the necessary OpenAI or Azure API keys
+  (OPENAI_API_KEY or AZURE_OPENAI_API_KEY)
+- Run unit tests: `make test`
+- Review coverage with `make coverage`
+- Interactively testing queries: `make demo`
+- Comparing to a validated set of questions and expected answers:
+  - Obtain or construct a JSON file of q/a pairs and install in testdata
+  - Run `make compare` (takes about 5-10 seconds per q/a pair)
+
+## Trademarks
+
+This project may contain trademarks or logos for projects, products, or services.
+Authorized use of Microsoft trademarks or logos is subject to and must follow
+[Microsoft's Trademark & Brand Guidelines](https://www.microsoft.com/en-us/legal/intellectualproperty/trademarks/usage/general).
+Use of Microsoft trademarks or logos in modified versions of this project
+must not cause confusion or imply Microsoft sponsorship.
+Any use of third-party trademarks or logos are subject to those third-party's policies.

typeagent_py-0.1.0/pyproject.toml

@@ -0,0 +1,57 @@
+[build-system]
+requires = ["setuptools>=67", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "typeagent-py"
+version = "0.1.0"
+description = "TypeAgent implements an agentic memory framework."
+authors = [
+  { name = "Guido van Rossum", email = "gvanrossum@microsoft.com" },
+  { name = "Steven Lucco" },
+  { name = "Umesh Madan" },
+  { name = "Guido van Rossum" },
+]
+requires-python = ">=3.12,<3.14"
+dependencies = [
+  "azure-identity>=1.22.0",
+  "black>=25.1.0",
+  "build>=1.2.2.post1",
+  "colorama>=0.4.6",
+  "coverage[toml]>=7.9.1",
+  "logfire>=4.1.0",
+  "mcp[cli]>=1.12.1",
+  "numpy>=2.2.6",
+  "openai>=1.81.0",
+  "opentelemetry-instrumentation-httpx>=0.57b0",
+  "pydantic>=2.11.4",
+  "pydantic-ai-slim[openai]>=0.5.0",
+  "pyright>=1.1.405",
+  "pytest>=8.3.5",
+  "pytest-asyncio>=0.26.0",
+  "pytest-mock>=3.14.0",
+  "python-dotenv>=1.1.0",
+  "typechat",
+]
+
+[project.urls]
+Homepage = "https://github.com/microsoft/TypeAgent/tree/main/python/ta"
+
+[tool.setuptools]
+# Needed so setuptools doesn't complain about testdata.
+packages = [
+    "typeagent",
+    "typeagent.aitools",
+    "typeagent.knowpro",
+    "typeagent.mcp",
+    "typeagent.podcasts",
+    "typeagent.storage",
+    "typeagent.storage.memory",
+    "typeagent.storage.sqlite",
+]
+
+[tool.pytest.ini_options]
+asyncio_default_fixture_loop_scope = "function"
+
+[tool.uv.sources]
+typechat = { git = "https://github.com/microsoft/TypeChat", subdirectory = "python" }

typeagent_py-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

typeagent_py-0.1.0/test/test_auth.py

@@ -0,0 +1,103 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+import time
+import pytest
+from pytest_mock import MockerFixture
+
+from typeagent.aitools.auth import AzureTokenProvider, get_shared_token_provider
+
+
+@pytest.fixture
+def mock_azure_token_provider(mocker: MockerFixture) -> AzureTokenProvider:
+    """Fixture to create a mocked AzureTokenProvider."""
+    mock_credential = mocker.patch(
+        "typeagent.aitools.auth.DefaultAzureCredential"
+    ).return_value
+    provider = AzureTokenProvider()
+    provider.credential = mock_credential
+    return provider
+
+
+def test_get_token_with_refresh(
+    mock_azure_token_provider: AzureTokenProvider, mocker: MockerFixture
+):
+    """Test that get_token refreshes the token when needed."""
+    mocker.patch.object(mock_azure_token_provider, "needs_refresh", return_value=True)
+    mock_refresh_token = mocker.patch.object(
+        mock_azure_token_provider, "refresh_token", return_value="new_token"
+    )
+
+    token = mock_azure_token_provider.get_token()
+
+    mock_azure_token_provider.needs_refresh.assert_called_once()  # type: ignore
+    mock_refresh_token.assert_called_once()
+    assert token == "new_token"
+
+
+def test_get_token_without_refresh(
+    mock_azure_token_provider: AzureTokenProvider, mocker: MockerFixture
+):
+    """Test that get_token uses the cached token when refresh is not needed."""
+    mocker.patch.object(mock_azure_token_provider, "needs_refresh", return_value=False)
+    mock_azure_token_provider.access_token = mocker.MagicMock(token="cached_token")
+
+    token = mock_azure_token_provider.get_token()
+
+    mock_azure_token_provider.needs_refresh.assert_called_once()  # type: ignore
+    assert token == "cached_token"
+
+
+def test_refresh_token(
+    mock_azure_token_provider: AzureTokenProvider, mocker: MockerFixture
+):
+    """Test that refresh_token retrieves a new token."""
+    mock_get_token = mocker.patch.object(
+        mock_azure_token_provider.credential,
+        "get_token",
+        return_value=mocker.MagicMock(
+            token="new_token", expires_on=int(time.time()) + 3600
+        ),
+    )
+
+    token = mock_azure_token_provider.refresh_token()
+
+    mock_get_token.assert_called_once_with(
+        "https://cognitiveservices.azure.com/.default"
+    )
+    assert token == "new_token"
+    assert mock_azure_token_provider.access_token is not None
+    assert mock_azure_token_provider.access_token.token == "new_token"
+
+
+def test_needs_refresh_no_token(mock_azure_token_provider: AzureTokenProvider):
+    """Test that needs_refresh returns True when no token is present."""
+    mock_azure_token_provider.access_token = None
+    assert mock_azure_token_provider.needs_refresh() is True
+
+
+def test_needs_refresh_expired_token(
+    mock_azure_token_provider: AzureTokenProvider, mocker: MockerFixture
+):
+    """Test that needs_refresh returns True when the token is about to expire."""
+    mock_azure_token_provider.access_token = mocker.MagicMock(
+        expires_on=int(time.time()) + 100
+    )
+    assert mock_azure_token_provider.needs_refresh() is True
+
+
+def test_needs_refresh_valid_token(
+    mock_azure_token_provider: AzureTokenProvider, mocker: MockerFixture
+):
+    """Test that needs_refresh returns False when the token is valid."""
+    mock_azure_token_provider.access_token = mocker.MagicMock(
+        expires_on=int(time.time()) + 3600
+    )
+    assert mock_azure_token_provider.needs_refresh() is False
+
+
+def test_get_shared_token_provider():
+    """Test that get_shared_token_provider returns a singleton instance."""
+    provider1 = get_shared_token_provider()
+    provider2 = get_shared_token_provider()
+    assert provider1 is provider2