swiftpy-framework 0.1.0__py3-none-any.whl

swiftpy/ai/__init__.py

@@ -0,0 +1,199 @@
+from __future__ import annotations
+
+from swiftpy.ai.agents import (
+    Agent,
+    AgentMemory,
+    AgentResponse,
+    AgentTask,
+    BaseAgent,
+    BaseMemoryBackend,
+    DatabaseMemoryBackend,
+    InMemoryMemoryBackend,
+    RedisMemoryBackend,
+)
+from swiftpy.ai.client import AI, AIClient, ai
+from swiftpy.ai.conversation import Conversation
+from swiftpy.ai.embeddings import (
+    EmbeddingsClient,
+    generate_deterministic_embedding,
+)
+from swiftpy.ai.exceptions import (
+    AgentLoopError,
+    AIAuthenticationError,
+    AIBudgetExceededError,
+    AIConfigurationError,
+    AIConnectionError,
+    AIContentModerationError,
+    AIError,
+    AIModuleDisabledError,
+    AIProviderError,
+    AIRateLimitError,
+)
+from swiftpy.ai.history_strategies import (
+    HistoryStrategy,
+    SlidingWindowStrategy,
+    TokenBudgetStrategy,
+)
+from swiftpy.ai.middleware import (
+    AIMiddleware,
+    AIMiddlewarePipeline,
+    AIRateLimitMiddleware,
+    ContentModerationMiddleware,
+    TokenBudgetMiddleware,
+)
+from swiftpy.ai.observability import (
+    AIObservabilityManager,
+    AIObservation,
+    BaseObservabilityDriver,
+    DatabaseObservabilityDriver,
+    FileObservabilityDriver,
+    InMemoryObservabilityDriver,
+    ObservationContext,
+    compute_prompt_hash,
+    get_observability,
+)
+from swiftpy.ai.pool import AIHttpPool
+from swiftpy.ai.pricing import (
+    calculate_token_cost,
+    register_model_pricing,
+)
+from swiftpy.ai.prompts import (
+    PromptLoader,
+    load_prompt,
+)
+from swiftpy.ai.providers.base import (
+    LLMRequest,
+    LLMResponse,
+    Message,
+    ProviderInterface,
+    TokenUsage,
+    ToolCall,
+    ToolResult,
+)
+from swiftpy.ai.rag import (
+    BaseReranker,
+    CohereReranker,
+    ContextFormatter,
+    CrossEncoderReranker,
+    NumberedListFormatter,
+    PlainConcatenationFormatter,
+    RAGPipelineBuilder,
+    RAGResult,
+    Retriever,
+    XMLBlocksFormatter,
+    rag,
+)
+from swiftpy.ai.redis_store import (
+    ConversationStore,
+    InMemoryConversationStore,
+    RedisConversationStore,
+)
+from swiftpy.ai.streaming import parse_sse_events
+from swiftpy.ai.tools import (
+    BaseTool,
+    CalculatorTool,
+    DateTimeTool,
+    FunctionTool,
+    ToolLoop,
+    WebSearchTool,
+    tool,
+)
+from swiftpy.ai.vectorstore import (
+    ChromaAdapter,
+    InMemoryVectorStore,
+    LanceDbAdapter,
+    PgVectorAdapter,
+    PineconeAdapter,
+    QdrantAdapter,
+    SearchResult,
+    VectorRecord,
+    VectorStore,
+)
+
+__all__ = [
+    "AI",
+    "AIAuthenticationError",
+    "AIBudgetExceededError",
+    "AIClient",
+    "AIConfigurationError",
+    "AIConnectionError",
+    "AIContentModerationError",
+    "AIError",
+    "AIHttpPool",
+    "AIMiddleware",
+    "AIMiddlewarePipeline",
+    "AIModuleDisabledError",
+    "AIObservabilityManager",
+    "AIObservation",
+    "AIProviderError",
+    "AIRateLimitError",
+    "AIRateLimitMiddleware",
+    "Agent",
+    "AgentLoopError",
+    "AgentMemory",
+    "AgentResponse",
+    "AgentTask",
+    "BaseAgent",
+    "BaseMemoryBackend",
+    "BaseObservabilityDriver",
+    "BaseReranker",
+    "BaseTool",
+    "CalculatorTool",
+    "ChromaAdapter",
+    "CohereReranker",
+    "ContentModerationMiddleware",
+    "ContextFormatter",
+    "Conversation",
+    "ConversationStore",
+    "CrossEncoderReranker",
+    "DatabaseMemoryBackend",
+    "DatabaseObservabilityDriver",
+    "DateTimeTool",
+    "EmbeddingsClient",
+    "FileObservabilityDriver",
+    "FunctionTool",
+    "HistoryStrategy",
+    "InMemoryConversationStore",
+    "InMemoryMemoryBackend",
+    "InMemoryObservabilityDriver",
+    "InMemoryVectorStore",
+    "LLMRequest",
+    "LLMResponse",
+    "LanceDbAdapter",
+    "Message",
+    "NumberedListFormatter",
+    "ObservationContext",
+    "PgVectorAdapter",
+    "PineconeAdapter",
+    "PlainConcatenationFormatter",
+    "PromptLoader",
+    "ProviderInterface",
+    "QdrantAdapter",
+    "RAGPipelineBuilder",
+    "RAGResult",
+    "RedisConversationStore",
+    "RedisMemoryBackend",
+    "Retriever",
+    "SearchResult",
+    "SlidingWindowStrategy",
+    "TokenBudgetMiddleware",
+    "TokenBudgetStrategy",
+    "TokenUsage",
+    "ToolCall",
+    "ToolLoop",
+    "ToolResult",
+    "VectorRecord",
+    "VectorStore",
+    "WebSearchTool",
+    "XMLBlocksFormatter",
+    "ai",
+    "calculate_token_cost",
+    "compute_prompt_hash",
+    "generate_deterministic_embedding",
+    "get_observability",
+    "load_prompt",
+    "parse_sse_events",
+    "rag",
+    "register_model_pricing",
+    "tool",
+]

swiftpy/ai/_deterministic.py

@@ -0,0 +1,40 @@
+from __future__ import annotations
+
+import hashlib
+import math
+
+
+def generate_deterministic_embedding(text: str, dimensions: int = 1536) -> list[float]:
+    """
+    Generate deterministic, normalized test embeddings using character n-grams and SHA-256 feature hashing.
+
+    This provides dependency-free embeddings for development and functional testing;
+    it is not a substitute for trained semantic embedding models.
+    """
+    if not text or not text.strip():
+        return [0.0] * dimensions
+
+    vec = [0.0] * dimensions
+    raw_words = [w.strip(".,;:!?\"'()[]{}").lower() for w in text.split() if w.strip()]
+
+    for word in raw_words:
+        tokens = [f"<{word}>"]
+        word_len = len(word)
+        for n in (3, 4, 5):
+            if word_len >= n:
+                for i in range(word_len - n + 1):
+                    tokens.append(word[i : i + n])
+
+        for tok in tokens:
+            digest = hashlib.sha256(tok.encode("utf-8")).digest()
+            for i in range(2):
+                idx = int.from_bytes(digest[i * 4 : (i + 1) * 4], "big") % dimensions
+                sign = 1.0 if digest[i] % 2 == 0 else -1.0
+                vec[idx] += sign
+
+    # Normalize vector to unit length (L2 norm)
+    norm = math.sqrt(sum(x * x for x in vec))
+    if norm > 0.0:
+        vec = [x / norm for x in vec]
+
+    return vec