Remove unused 'model' variable and logging statements (#2228)

This pull request removes the unused 'model' variable and logging
statements from the codebase.
This commit is contained in:
Stan Girard 2024-02-20 02:14:02 -08:00 committed by GitHub
parent 9658bb8742
commit 956ffbb5e0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 1 additions and 9 deletions

View File

@ -94,7 +94,6 @@ class CompositeBrainQA(
return KnowledgeBrainQA(
brain_id=str(brain.id),
chat_id=self.chat_id,
model=self.model,
max_tokens=self.max_tokens,
temperature=self.temperature,
streaming=self.streaming,

View File

@ -177,7 +177,7 @@ class KnowledgeBrainQA(BaseModel, QAInterface):
self.models_settings = self.user_usage.get_model_settings()
self.increase_usage_user()
self.knowledge_qa = QuivrRAG(
model=self.brain.model,
model=self.brain.model if self.brain.model else self.model,
brain_id=brain_id,
chat_id=chat_id,
streaming=streaming,

View File

@ -45,17 +45,12 @@ class BrainsUsers(BrainsUsersInterface):
for item in response.data:
integration_logo_url = ""
if item["brains"]["brain_type"] == "integration":
logger.info("Integration brain")
logger.info(item["brains"])
if "integrations_user" in item["brains"]:
for integration_user in item["brains"]["integrations_user"]:
if "integrations" in integration_user:
integration_logo_url = integration_user["integrations"][
"integration_logo_url"
]
logger.info(
"Integration logo url: " + str(integration_logo_url)
)
user_brains.append(
MinimalUserBrainEntity(

View File

@ -81,7 +81,6 @@ class BrainfulChat(ChatInterface):
)
return APIBrainQA(
chat_id=chat_id,
model=model,
temperature=temperature,
brain_id=str(brain.brain_id),
streaming=streaming,
@ -104,7 +103,6 @@ class BrainfulChat(ChatInterface):
if integration_class:
return integration_class(
chat_id=chat_id,
model=model,
temperature=temperature,
brain_id=str(brain.brain_id),
streaming=streaming,