several fixes before initial release

Author: fswair

.github/workflows/publish.yml

@@ -0,0 +1,32 @@
+name: Publish to TestPyPI with uv
+
+on:
+  push:
+    tags:
+      - 'v*'
+
+jobs:
+  publish:
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v4
+
+    - name: Set up Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.11'
+
+    - name: Install uv
+      run: pip install uv
+
+    - name: Install dependencies
+      run: uv pip install setuptools wheel twine
+
+    - name: Build package
+      run: python setup.py sdist bdist_wheel
+
+    - name: Publish to TestPyPI
+      env:
+        PYPI_TOKEN: ${{ secrets.TEST_PYPI_TOKEN }}
+      run: twine upload dist/* --repository testpypi --username __token__ --password $PYPI_TOKEN

VERSION

@@ -0,0 +1 @@
+1.28.3

requirements.txt

@@ -5,6 +5,7 @@ tinydb
 fasta2a
 attrobj
 fastapi
+logfire
 uvicorn
 pydantic
 aiocache

setup.py

@@ -0,0 +1,46 @@
+from setuptools import setup
+from pathlib import Path
+
+parent = Path(__file__).parent
+
+description = (parent / "README.md").read_text()
+requirements = (parent / "requirements.txt").read_text().splitlines()
+
+setup(
+    name="venai",
+    version="1.28.3",
+    author="Mert Sirakaya",
+    author_email="contact@tomris.dev",
+    maintainer="Mert Sirakaya",
+    maintainer_email="contact@tomris.dev",
+    description="VenusAI is a secure and extensible Agent framework built for modern AI applications.",
+    packages=["venus", "venus.helpers", "venus.helpers.sandbox", "venus.models"],
+    fullname="VenusAI",
+    install_requires=requirements,
+    long_description=description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/VenusAgent/VenusAI",
+    classifiers=[
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
+        "Programming Language :: Python :: 3.12",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+    ],
+    python_requires=">=3.10",
+    include_package_data=True,
+    license="MIT",
+    keywords="safety, ai, agent, llm, agentic ai, pydantic, pydantic-ai, self-healing, sandbox, ai agent",
+    project_urls={
+        "Documentation": "https://venai.tech/docs",
+        "Source": "https://github.com/VenusAgent/VenusAI",
+        "Tracker": "https://github.com/VenusAgent/VenusAI/issues",
+    },
+    entry_points={
+        "console_scripts": [
+            "venus=venus.__main__:main",
+            "venai=venus.__main__:main",
+        ],
+    },
+    
+)

tests/empty

@@ -0,0 +1 @@
+silence is golden

venus/__init__.py

@@ -5,7 +5,7 @@
 """
 
 from .agent import Tool, Venus, VenusCode
-from .decorators import tool, mcp_tool, safe_call, autofix
+from .decorators import autofix, mcp_tool, safe_call, tool
 from .errors import ErrorDict
 from .helpers import e2b, tools
 
@@ -21,3 +21,5 @@
     "e2b",
     "tools",
 ]
+
+__version__ = "1.28.3"

venus/__main__.py

@@ -6,6 +6,7 @@
 
 import importlib
 import os
+import sys
 from typing import Any, cast
 
 import click
@@ -19,6 +20,8 @@
 
 vc = VenusConsole()
 
+sys.path.insert(0, os.getcwd())
+
 
 def attr_resolve(obj: Any, path: str):
     """
@@ -38,12 +41,12 @@ def attr_resolve(obj: Any, path: str):
     return obj
 
 
-def start_chat(app: Venus, deps: None = None):
+def start_chat(app: Venus, deps: None = None, prog_name: str | None = None):
     """
     Start the chat interface.
     """
-    name = app.name or "venus"
-    app.to_cli_sync(deps=deps, prog_name=name)
+    prog_name = prog_name or app.name or "venus"
+    app.to_cli_sync(deps=deps, prog_name=prog_name)
 
 
 @click.group()
@@ -57,7 +60,8 @@ def main():
 
 @main.command()
 @click.argument("app_string")
-def chat(app_string: str):
+@click.option("--name", default="venus", help="Name of the chat assistant")
+def chat(app_string: str, name: str):
     """Start the chat interface."""
     try:
         module_name, app_name = app_string.split(":", maxsplit=1)
@@ -70,7 +74,7 @@ def chat(app_string: str):
         else:
             app = getattr(module, app_name)
 
-        start_chat(app)
+        start_chat(app, prog_name=name)
 
     except ValueError:
         vc.print_exception(show_locals=True)
@@ -175,7 +179,6 @@ def serve(app: str, host: str, port: int, auto: bool, env: str, plugin: str = No
     try:
         module, app = app.split(":")
         app = app.removesuffix(".api")
-
         module_instance = importlib.import_module(module)
         app_instance = attr_resolve(module_instance, app)
 
@@ -221,7 +224,7 @@ def serve(app: str, host: str, port: int, auto: bool, env: str, plugin: str = No
             vc.fail(
                 "[cyan]App.tools_http_api()[/cyan] "
                 "[bold yellow]not called yet.[/bold yellow]\n"
-                "[bold green]Hint:[/bold green] Pass [italic]--auto[/italic] to use registered tools of Agent automatically."
+                "[bold green]Hint:[/bold green]     Pass [italic]--auto[/italic] to use registered tools of Agent automatically."
             )
             return
 

venus/_decorator_utils.py

@@ -10,29 +10,16 @@
 from datetime import datetime
 from pathlib import Path
 from types import TracebackType
-from typing import (
-    Any,
-    Callable,
-    ParamSpec,
-    TypeVar,
-    Union,
-    cast,
-    get_origin,
-    get_type_hints,
-)
+from typing import (Any, Callable, ParamSpec, TypeVar, Union, cast,
+                    get_type_hints)
 
 from attrobj import Object
 from pydantic_ai import StructuredDict
 
 from venus.mock_types import Autofix
 
-from .errors import (
-    ContextParamDuplicated,
-    ErrorDict,
-    InvalidContextParam,
-    InvalidDependencyParam,
-    MainBlockNotFound,
-)
+from .errors import (ContextParamDuplicated, ErrorDict, InvalidContextParam,
+                     InvalidDependencyParam, MainBlockNotFound)
 from .logger import VenusConsole
 from .schemas import FixFuncResult
 from .settings import settings

venus/agent.py

@@ -1,83 +1,51 @@
-import json
-import types
-import tinydb
-import inspect
-import logfire
 import asyncio
+import inspect
+import json
 import logging
-import uvicorn
-
-from pathlib import Path
-from datetime import datetime
-from dotenv import load_dotenv
+import os
+import types
 from collections import defaultdict
-
 from contextlib import asynccontextmanager
+from datetime import datetime
+from pathlib import Path
+from typing import (Any, Callable, Dict, Generic, Literal, Optional, Sequence,
+                    TypeVar, Union, cast, get_type_hints, overload)
 
-
-from .settings import Settings
-from .logger import VenusConsole
-from .schemas import DoesNeedFix, FixFuncResult
-from .prompts import CODING_PROMPT
-from .helpers import tools, time_diff_prettify
-from .permissions import Permissions, get_allowed_tools
-from ._module_utils import import_module
-from ._decorator_utils import (
-    extract_function_body,
-    has_context_param,
-)
-from .decorators import safe_call, is_context_tool
-
-from typing import (
-    Any,
-    Dict,
-    Union,
-    Literal,
-    TypeVar,
-    Generic,
-    Callable,
-    Sequence,
-    cast,
-    overload,
-    Optional,
-    get_type_hints,
-)
-
+import logfire
+import tinydb
+import uvicorn
+from dotenv import load_dotenv
 from fastapi import APIRouter as Router
 from fastapi import FastAPI as Server
-
 from pydantic_ai import Agent, EndStrategy
-from pydantic_ai.agent.abstract import EventStreamHandler
 from pydantic_ai import _system_prompt as _system_prompt
-from pydantic_ai._run_context import AgentDepsT
-from pydantic_ai.settings import ModelSettings
-from pydantic_ai.output import OutputSpec, OutputDataT
 from pydantic_ai._agent_graph import HistoryProcessor
-from pydantic_ai.toolsets import ToolsetFunc
-from pydantic_ai.agent import (
-    InstrumentationSettings,
-)
+from pydantic_ai._run_context import AgentDepsT
+from pydantic_ai.agent import InstrumentationSettings
+from pydantic_ai.agent.abstract import EventStreamHandler
+from pydantic_ai.builtin_tools import AbstractBuiltinTool
 from pydantic_ai.mcp import MCPServer
 from pydantic_ai.models import KnownModelName
 from pydantic_ai.models.openai import Model
-from pydantic_ai.tools import Tool, ToolFuncContext, ToolFuncEither, ToolFuncPlain
-
-from .types import FuncParams, ToolsPrepareFunc, _EnableFeature, Deps
-
-from pydantic_ai.tools import ToolParams
-from pydantic_ai.toolsets import AbstractToolset
-from pydantic_ai.builtin_tools import AbstractBuiltinTool
-
-from .errors import (
-    ErrorDict,
-    ExecutionNotAllowed,
-    InvalidFunction,
-    InvalidTool,
-    InvalidTools,
-    InvalidContextParam,
-)
+from pydantic_ai.output import OutputDataT, OutputSpec
+from pydantic_ai.settings import ModelSettings
+from pydantic_ai.tools import (Tool, ToolFuncContext, ToolFuncEither,
+                               ToolFuncPlain, ToolParams)
+from pydantic_ai.toolsets import AbstractToolset, ToolsetFunc
 
 from . import decorators
+from ._decorator_utils import extract_function_body, has_context_param
+from ._module_utils import import_module
+from .decorators import is_context_tool, safe_call
+from .errors import (ErrorDict, ExecutionNotAllowed, InvalidContextParam,
+                     InvalidFunction, InvalidTool, InvalidTools)
+from .helpers import time_diff_prettify, tools
+from .logger import VenusConsole
+from .permissions import Permissions, get_allowed_tools
+from .prompts import CODING_PROMPT
+from .schemas import DoesNeedFix, FixFuncResult
+from .settings import Settings
+from .types import Deps, FuncParams, ToolsPrepareFunc, _EnableFeature
 
 """
 Agent module for building and configuring an agent with HTTP client support.
@@ -448,7 +416,7 @@ def build_agent(
             )
         self._agent_built = True
         if load_env:
-            load_dotenv(override=override_env)
+            load_dotenv(Path(os.getcwd()) / ".env", override=override_env)
 
         model = settings.model_name
         system_prompt = settings.system_prompt.format(

venus/caching.py

@@ -77,15 +77,8 @@ def decorator(
                 ) from e
         else:
             try:
-                from cachetools import (
-                    FIFOCache,
-                    LFUCache,
-                    LRUCache,
-                    RRCache,
-                    TLRUCache,
-                    TTLCache,
-                    cached,
-                )
+                from cachetools import (FIFOCache, LFUCache, LRUCache, RRCache,
+                                        TLRUCache, TTLCache, cached)
                 from cachetools.keys import hashkey
 
                 cachekey = (