feat(integrations): add Hermes provider with shared memory utilities

2026-09-18 13:21:43 +05:30
parent 84bf468176
commit 4284ef0d2a
40 changed files with 4168 additions and 954 deletions
@@ -0,0 +1,98 @@
+"""OSS provider definitions for LLM, embedder, and vector store."""
+
+from __future__ import annotations
+
+import os
+from typing import Any
+
+from hermes_constants import get_hermes_home
+
+LLM_PROVIDERS: dict[str, dict[str, Any]] = {
+    "openai": {
+        "label": "OpenAI",
+        "needs_key": True,
+        "env_var": "OPENAI_API_KEY",
+        "default_model": "gpt-5-mini",
+        "base_url_key": "openai_base_url",
+    },
+    "ollama": {
+        "label": "Ollama (local)",
+        "needs_key": False,
+        "default_model": "llama3.1:8b",
+        "default_url": "http://localhost:11434",
+        "base_url_key": "ollama_base_url",
+        "pip_dep": "ollama",
+    },
+}
+
+EMBEDDER_PROVIDERS: dict[str, dict[str, Any]] = {
+    "openai": {
+        "label": "OpenAI",
+        "needs_key": True,
+        "env_var": "OPENAI_API_KEY",
+        "default_model": "text-embedding-3-small",
+        "base_url_key": "openai_base_url",
+        "dims": 1536,
+    },
+    "ollama": {
+        "label": "Ollama (local)",
+        "needs_key": False,
+        "default_model": "nomic-embed-text",
+        "default_url": "http://localhost:11434",
+        "base_url_key": "ollama_base_url",
+        "dims": 768,
+        "pip_dep": "ollama",
+    },
+}
+
+VECTOR_PROVIDERS: dict[str, dict[str, Any]] = {
+    # Resolved lazily (see ``vector_default_config``): the profile home is a ContextVar at call time,
+    # not an import-time constant, and ``~/.hermes`` is wrong on Windows and under profiles.
+    "qdrant": {
+        "label": "Qdrant",
+        "default_config": {"path": lambda: str(get_hermes_home() / "mem0_qdrant")},
+        "pip_dep": "qdrant-client",
+    },
+    "pgvector": {
+        "label": "PGVector",
+        "default_config": {
+            "host": "localhost",
+            "port": 5432,
+            "user": os.getenv("USER", "postgres"),
+            "dbname": "postgres",
+        },
+        "pip_dep": "psycopg2-binary",
+    },
+}
+
+KNOWN_DIMS: dict[str, int] = {
+    "text-embedding-3-small": 1536,
+    "text-embedding-3-large": 3072,
+    "text-embedding-ada-002": 1536,
+    "nomic-embed-text": 768,
+}
+
+
+def vector_default_config(provider_id: str) -> dict[str, Any]:
+    """A vector store's ``default_config`` with callable defaults resolved for the active profile."""
+    return {k: (v() if callable(v) else v) for k, v in VECTOR_PROVIDERS[provider_id]["default_config"].items()}
+
+
+SECTION_REGISTRIES = (("llm", LLM_PROVIDERS), ("embedder", EMBEDDER_PROVIDERS), ("vector_store", VECTOR_PROVIDERS))
+
+
+def validate_oss_config(oss_config: dict) -> list[str]:
+    """Validate an OSS config dict. Returns list of error strings (empty = valid)."""
+    errors: list[str] = []
+    for section, registry in SECTION_REGISTRIES:
+        block = oss_config.get(section)
+        if not block or not isinstance(block, dict):
+            errors.append(f"Missing required section: {section}")
+        elif block.get("provider", "") not in registry:
+            errors.append(
+                f"Unknown {section} provider '{block.get('provider', '')}'. Valid: {', '.join(registry.keys())}"
+            )
+    vs = oss_config.get("vector_store", {})
+    if vs.get("provider") == "pgvector" and not vs.get("config", {}).get("user"):
+        errors.append("PGVector requires 'user' in vector_store.config")
+    return errors