Files
replypier/backend/app/repositories/jira.py
T
WangShen007 54b949b392 chore(brand): rename the product to ReplyPier
The upstream project name is gone from code, assets, docs and build
output: brand strings, domains (replypier.site and its app/api/docs/help
subdomains), env prefixes, image and network names, the widget's public
JS API (window.ReplyPier, replypierId/BaseUrl/Config/Token), CSS tokens
(--rp-*), file and directory names, and the WordPress plugin package.

- widget loader is now webclient/replypier.min.js; both bundles rebuilt
  from source (backend/assets/widget.{js,css}, frontend/public/webclient)
- NOTICE, llms.txt, CONTRIBUTING, HELP_CENTER_INFRA and config.dev.js
  rewritten rather than string-swapped; the demo config no longer ships
  the previous vendor's Firebase/Razorpay/Google keys
- new migration replypier_ai_model_001 adds REPLYPIER to the managed
  AIModelType and rewrites pre-rename rows
- local compose project, database, role and volume renamed; LOCAL_DEPLOY
  documents carrying the old local data over
2026-10-26 00:51:03 +08:00

116 lines
4.6 KiB
Python

"""
Copyright 2024-2026 ReplyPier
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
from sqlalchemy.orm import Session, joinedload
from sqlalchemy import cast, String, func
from typing import Optional, Dict, Any
from app.models.jira import AgentJiraConfig
from app.models.agent import Agent
from uuid import UUID
from app.core.logger import get_logger
from app.models.schemas.jira import AgentWithJiraConfig
import json
logger = get_logger(__name__)
class JiraRepository:
def __init__(self, db: Session):
self.db = db
def get_agent_with_jira_config(self, agent_id: str) -> Optional[AgentWithJiraConfig]:
"""
Get agent with its Jira configuration in a single optimized query.
Returns an AgentWithJiraConfig object that can be accessed with dot notation.
"""
try:
if isinstance(agent_id, str):
agent_id = UUID(agent_id)
# First get the agent
agent = self.db.query(Agent).options(
joinedload(Agent.groups),
joinedload(Agent.organization)
).filter(
Agent.id == agent_id
).first()
if not agent:
return None
# Then get the Jira config
jira_config = self.db.query(AgentJiraConfig).filter(
AgentJiraConfig.agent_id == str(agent.id)
).first()
# Parse tools JSON if it exists
tools_list = []
if agent.tools:
try:
if isinstance(agent.tools, str):
# Parse JSON string
parsed = json.loads(agent.tools)
tools_list = parsed if isinstance(parsed, list) else []
elif isinstance(agent.tools, list):
# Already a list
tools_list = agent.tools
elif isinstance(agent.tools, dict):
# If it's a dict (empty or not), convert to empty list
logger.warning(f"Agent {agent.id} has tools as dict instead of list, converting to empty list")
tools_list = []
else:
# Unknown type
logger.warning(f"Agent {agent.id} has tools of unexpected type: {type(agent.tools)}")
tools_list = []
except (json.JSONDecodeError, TypeError) as e:
logger.warning(f"Error parsing tools for agent {agent.id}: {str(e)}")
tools_list = []
# Create an AgentWithJiraConfig object
agent_data = AgentWithJiraConfig(
id=agent.id,
name=agent.name,
display_name=agent.display_name,
description=agent.description,
instructions=agent.instructions,
tools=tools_list,
agent_type=agent.agent_type,
is_default=agent.is_default,
is_active=agent.is_active,
organization_id=agent.organization_id,
transfer_to_human=agent.transfer_to_human,
ask_for_rating=agent.ask_for_rating,
ticketing_enabled=agent.ticketing_enabled,
topic_scope=agent.topic_scope,
guardrail_prompt=agent.guardrail_prompt,
guardrail_enabled=agent.guardrail_enabled,
groups=agent.groups,
organization=agent.organization,
knowledge=[], # Empty list as default
jira_enabled=False,
jira_project_key=None,
jira_issue_type_id=None
)
# Add Jira configuration if it exists
if jira_config:
agent_data.jira_enabled = jira_config.enabled
agent_data.jira_project_key = jira_config.project_key
agent_data.jira_issue_type_id = jira_config.issue_type_id
return agent_data
except Exception as e:
logger.error(f"Error getting agent with Jira config: {str(e)}")
return None