Tests: fix local and remote llm gateway tests

.github/workflows/langchain_ci.yml

@@ -24,6 +24,10 @@ jobs:
     uses: ./.github/workflows/test.yml
     with:
       folder_path: "."
+    secrets:
+      UIPATH_URL: ${{ secrets.UIPATH_URL }}
+      UIPATH_CLIENT_ID: ${{ secrets.UIPATH_CLIENT_ID }}
+      UIPATH_CLIENT_SECRET: ${{ secrets.UIPATH_CLIENT_SECRET }}
 
   build:
     needs: [lint, test]

.github/workflows/test.yml

@@ -2,8 +2,14 @@ name: Reusable Test Workflow
 
 on:
   workflow_call:
+    secrets:
+      UIPATH_URL:
+        required: true
+      UIPATH_CLIENT_ID:
+        required: true
+      UIPATH_CLIENT_SECRET:
+        required: true
     inputs:
-
       folder_path:
         description: 'The folder path to run actions in'
         required: true
@@ -44,6 +50,4 @@ jobs:
           UIPATH_URL: ${{ secrets.UIPATH_URL }}
           UIPATH_CLIENT_ID: ${{ secrets.UIPATH_CLIENT_ID }}
           UIPATH_CLIENT_SECRET: ${{ secrets.UIPATH_CLIENT_SECRET }}
-          UIPATH_ORGANIZATION_ID: ${{ vars.UIPATH_ORGANIZATION_ID }}
-          UIPATH_TENANT_ID: ${{ vars.UIPATH_TENANT_ID }}
 

.vscode/settings.json

@@ -21,7 +21,7 @@
     "titleBar.inactiveBackground": "#0099cc"
   },
   "python.testing.pytestArgs": [
-    "sdk"
+    "tests"
   ],
   "python.testing.unittestEnabled": false,
   "python.testing.pytestEnabled": true

tests/conftest.py

@@ -11,11 +11,30 @@
 from langchain_community.cache import SQLiteCache
 
 from uipath_langchain.embeddings import UiPathOpenAIEmbeddings
-from uipath_langchain.utils._settings import uipath_cached_paths_settings
+from uipath_langchain.utils._settings import UiPathCachedPathsSettings
+
+test_cache_settings = UiPathCachedPathsSettings(
+    CACHED_COMPLETION_DB="tests/llm_cache/tests_uipath_cache.sqlite",
+    CACHED_EMBEDDINGS_DIR="tests/llm_cache/cached_embeddings",
+)
+
+
+def get_from_uipath_url():
+    try:
+        url = os.getenv("UIPATH_URL")
+        if url:
+            return "/".join(url.split("/", 3)[:3])
+    except Exception:
+        return None
+    return None
 
 
 def get_token():
-    url_get_token = f"{env.get('UIPATH_URL', '').rstrip('/')}/identity_/connect/token"
+    url_get_token = f"{get_from_uipath_url().rstrip('/')}/identity_/connect/token"
+
+    os.environ["UIPATH_REQUESTING_PRODUCT"] = "uipath-python-sdk"
+    os.environ["UIPATH_REQUESTING_FEATURE"] = "langgraph-agent"
+    os.environ["UIPATH_TESTS_CACHE_LLMGW"] = "true"
 
     token_credentials = {
         "client_id": env.get("UIPATH_CLIENT_ID"),
@@ -49,7 +68,7 @@ def cached_llmgw_calls() -> Generator[Optional[SQLiteCache], None, None]:
         yield None
     else:
         logging.info("Setting up LLMGW cache")
-        db_path = uipath_cached_paths_settings.cached_completion_db
+        db_path = test_cache_settings.cached_completion_db
         os.makedirs(os.path.dirname(db_path), exist_ok=True)
         cache = SQLiteCache(database_path=db_path)
         set_llm_cache(cache)
@@ -68,7 +87,7 @@ def cached_embedder() -> Generator[Optional[CacheBackedEmbeddings], None, None]:
         embedder = CacheBackedEmbeddings.from_bytes_store(
             underlying_embeddings=UiPathOpenAIEmbeddings(model=model),
             document_embedding_cache=LocalFileStore(
-                uipath_cached_paths_settings.cached_embeddings_dir
+                test_cache_settings.cached_embeddings_dir
             ),
             namespace=model,
         )