Rm unused import and var

pull/22108/head
Yongtao Huang 11 months ago
parent a176533d6e
commit 263f6dfc05

@ -4,7 +4,6 @@ import time
from core.rag.datasource.vdb.couchbase.couchbase_vector import CouchbaseConfig, CouchbaseVector from core.rag.datasource.vdb.couchbase.couchbase_vector import CouchbaseConfig, CouchbaseVector
from tests.integration_tests.vdb.test_vector_store import ( from tests.integration_tests.vdb.test_vector_store import (
AbstractVectorTest, AbstractVectorTest,
get_example_text,
setup_mock_redis, setup_mock_redis,
) )

@ -1,7 +1,6 @@
from core.rag.datasource.vdb.matrixone.matrixone_vector import MatrixoneConfig, MatrixoneVector from core.rag.datasource.vdb.matrixone.matrixone_vector import MatrixoneConfig, MatrixoneVector
from tests.integration_tests.vdb.test_vector_store import ( from tests.integration_tests.vdb.test_vector_store import (
AbstractVectorTest, AbstractVectorTest,
get_example_text,
setup_mock_redis, setup_mock_redis,
) )

@ -5,7 +5,6 @@ import psycopg2 # type: ignore
from core.rag.datasource.vdb.opengauss.opengauss import OpenGauss, OpenGaussConfig from core.rag.datasource.vdb.opengauss.opengauss import OpenGauss, OpenGaussConfig
from tests.integration_tests.vdb.test_vector_store import ( from tests.integration_tests.vdb.test_vector_store import (
AbstractVectorTest, AbstractVectorTest,
get_example_text,
setup_mock_redis, setup_mock_redis,
) )

@ -1,7 +1,6 @@
from core.rag.datasource.vdb.pyvastbase.vastbase_vector import VastbaseVector, VastbaseVectorConfig from core.rag.datasource.vdb.pyvastbase.vastbase_vector import VastbaseVector, VastbaseVectorConfig
from tests.integration_tests.vdb.test_vector_store import ( from tests.integration_tests.vdb.test_vector_store import (
AbstractVectorTest, AbstractVectorTest,
get_example_text,
setup_mock_redis, setup_mock_redis,
) )

@ -113,8 +113,6 @@ def test_execute_llm(flask_req_ctx):
}, },
) )
credentials = {"openai_api_key": os.environ.get("OPENAI_API_KEY")}
# Create a proper LLM result with real entities # Create a proper LLM result with real entities
mock_usage = LLMUsage( mock_usage = LLMUsage(
prompt_tokens=30, prompt_tokens=30,

@ -14,9 +14,7 @@ from core.variables import (
ArrayStringVariable, ArrayStringVariable,
FloatVariable, FloatVariable,
IntegerVariable, IntegerVariable,
ObjectSegment,
SecretVariable, SecretVariable,
SegmentType,
StringVariable, StringVariable,
) )
from core.variables.exc import VariableError from core.variables.exc import VariableError

Loading…
Cancel
Save