Skip to content

Commit

Permalink
resolve lint
Browse files Browse the repository at this point in the history
  • Loading branch information
tianjing-li committed Jan 10, 2025
1 parent 808e854 commit efe5420
Show file tree
Hide file tree
Showing 6 changed files with 6 additions and 13 deletions.
2 changes: 1 addition & 1 deletion src/backend/routers/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,7 @@ async def create_agent(

deployment_db = default_deployment_db
model_db = default_model_db

try:
agent_data = AgentModel(
name=agent.name,
Expand Down
10 changes: 5 additions & 5 deletions src/backend/services/deployment.py
Original file line number Diff line number Diff line change
@@ -1,15 +1,15 @@
"""
This module handles backend operations related to deployments, which define how to interact
with an LLM.
This module handles backend operations related to deployments, which define how to interact
with an LLM.
New deployments are created by subclassing BaseDeployment and implementing required
New deployments are created by subclassing BaseDeployment and implementing required
methods in the model_deployments directory.
Deployments can be configured in two ways: via configuration.yaml or environment variables
using the .env file, or dynamically in the database using the deployment_crud module. This
using the .env file, or dynamically in the database using the deployment_crud module. This
service abstracts these methods, ensuring higher layers remain unaffected by configuration details.
Each deployment is assumed to use either configuration files or the database, with database
Each deployment is assumed to use either configuration files or the database, with database
configurations taking precedence.
"""

Expand Down
3 changes: 0 additions & 3 deletions src/backend/tests/integration/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,6 @@
from backend.database_models.deployment import Deployment
from backend.database_models.model import Model
from backend.main import app, create_app

# from backend.schemas.deployment import DeploymentDefinition
from backend.schemas.organization import Organization
from backend.schemas.user import User
from backend.tests.unit.factories import get_factory
Expand Down Expand Up @@ -184,7 +182,6 @@ def mock_available_model_deployments(request):
MockSageMakerDeployment,
)

# is_available_values = getattr(request, "param", {})
MOCKED_DEPLOYMENTS = {
MockCohereDeployment.name(): MockCohereDeployment,
MockAzureDeployment.name(): MockAzureDeployment,
Expand Down
1 change: 0 additions & 1 deletion src/backend/tests/integration/routers/test_chat.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,7 +211,6 @@ def test_streaming_chat_with_agent_tools_and_empty_request_tools(
"/v1/chat-stream",
headers={
"User-Id": agent.user.id,
# "Deployment-Name": agent.deployment,
"Deployment-Name": CohereDeployment.name(),
},
json={
Expand Down
1 change: 0 additions & 1 deletion src/backend/tests/unit/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,6 @@ def mock_available_model_deployments(request):
MockSingleContainerDeployment,
)

# is_available_values = getattr(request, "param", {})
MOCKED_DEPLOYMENTS = {
MockCohereDeployment.name(): MockCohereDeployment,
MockAzureDeployment.name(): MockAzureDeployment,
Expand Down
2 changes: 0 additions & 2 deletions src/backend/tests/unit/crud/test_document.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,6 @@
from backend.database_models.document import Document
from backend.tests.unit.factories import get_factory

# from backend.schemas.document import UpdateDocument


@pytest.fixture(autouse=True)
def conversation(session, user):
Expand Down

0 comments on commit efe5420

Please sign in to comment.