Added updated wheel packages

Signed-off-by: Yogesh <[email protected]>
yogeshmpandey · Jan 8, 2025 · 85c796c · 85c796c
1 parent c776d3b
commit 85c796c
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 7 deletions.
diff --git a/comps/integrations/langchain/dist/langchain_opea-0.1.0-py3-none-any.whl b/comps/integrations/langchain/dist/langchain_opea-0.1.0-py3-none-any.whl
diff --git a/comps/integrations/langchain/dist/langchain_opea-0.1.0.tar.gz b/comps/integrations/langchain/dist/langchain_opea-0.1.0.tar.gz
diff --git a/comps/integrations/langchain/langchain_opea/chat_models.py b/comps/integrations/langchain/langchain_opea/chat_models.py
@@ -44,10 +44,7 @@ class ChatOPEA(BaseChatOpenAI):  # type: ignore[override]
     """
     opea_api_base: str = Field(default="https://localhost:9009/v1/")
     """Base URL path for API requests."""
-
-    openai_api_key: Optional[SecretStr] = None
-    # openai_api_base: Optional[str] = None
-
+
     model_config = ConfigDict(
         populate_by_name=True,
     )

diff --git a/tests/integrations/test_integration_langchain_opea.sh b/tests/integrations/test_integration_langchain_opea.sh
@@ -34,6 +34,7 @@ function start_service() {
     docker run -d --name="test-comps-integration-llm-tgi-endpoint" -p $tgi_endpoint_port:80 -v ~/.cache/huggingface/hub:/data --shm-size 1g -e HF_TOKEN=${HF_TOKEN} ghcr.io/huggingface/text-generation-inference:2.1.0 --model-id ${hf_llm_model} --max-input-tokens 1024 --max-total-tokens 2048
     # check whether tgi is fully ready
     n=0
+    rm $LOG_PATH/test-comps-vllm-service.log
     until [[ "$n" -ge 100 ]] || [[ $ready == true ]]; do
         docker logs test-comps-integration-llm-tgi-endpoint >> $LOG_PATH/test-comps-vllm-service.log
         n=$((n+1))
@@ -81,18 +82,18 @@ function validate_package(){
     cd "$WORKPATH/GenAIComps/comps/integrations/langchain/"
     poetry run pytest --asyncio-mode=auto tests/
     if [ $? -ne 0 ]; then
-        echo "Package installation fail"
+        echo "Package Tests failed"
         exit 1
     else
-        echo "Package installation successful"
+        echo "Package Tests successful"
     fi
 
 }
 
 
 function remove_integration_package(){
     cd "$WORKPATH/GenAIComps/comps/integrations/langchain/"
-    pip uninstall langchain-opea
+    pip uninstall -y langchain-opea
     if [ $? -ne 0 ]; then
         echo "Package removal fail"
         exit 1