Try to preload tiktoken

Author: TimPansino

tests/mlmodel_langchain/conftest.py

@@ -16,6 +16,7 @@
 import os
 
 import pytest
+import tiktoken
 from langchain_core.messages.ai import AIMessage
 from langchain_core.messages.tool import ToolMessage
 from langchain_openai import ChatOpenAI, OpenAIEmbeddings
@@ -76,6 +77,8 @@
     ]
 )
 
+TIKTOKEN_ENCODINGS = ["cl100k_base"]
+
 
 # Intercept outgoing requests and log to file for mocking
 EXPECTED_AGENT_RESPONSE = "Hello!"
@@ -108,6 +111,13 @@ def embedding_openai_client(openai_clients):
     return embedding_client
 
 
+@pytest.fixture(scope="session")
+def load_tiktoken_encodings():
+    """Load tiktoken encodings before tests run to avoid issues with VCR blocking the network."""
+    for encoding in TIKTOKEN_ENCODINGS:
+        tiktoken.get_encoding(encoding)
+
+
 @pytest.fixture
 def chat_openai_client(openai_clients):
     chat_client, _ = openai_clients