visibility_service.cpython-311.pyc•16.8 kB
�
� �hF2 � � � d dl Z d dlmZmZ d dlmZmZmZmZ d dlm Z m
Z
d dlmZm
Z
d dlmZmZmZmZmZ d dlmZmZ d dlmZ G d � d
� � Z e� � ZdS )� N)�datetime� timedelta)�List�Dict�Any�Optional)�Session�select)�engine�ensure_user_exists)�User�TrackedQuery�Run�RunItem� Aggregate)�visibility_workflow�VisibilityState)�auto_detect_brandsc
�< � e Zd Zd� Zddededeee deeef fd�Z dedeeef fd�Z
dded ed
eee deeef fd�Zdded ed
edeeef fd�Z
dded ededee deeef f
d�Zdedeeef fd�ZdS )�VisibilityServicec � � ddg| _ d S )N�azure_openai�
perplexity)�default_platforms)�selfs �P/Users/kulraj/agent09/mcp-starter/mcp-bearer-token/../data/visibility_service.py�__init__zVisibilityService.__init__ s � �"0�,�!?����� N�phone�query�competitors�returnc � �� K � t |� � t t � � 5 }|� t t
� � � t
j |k t
j |k � � � � � � � }|r]|�E||_
|� |� � |� � � |�
|� � d|j d�cddd� � S t |||pg �� � }|� |� � |� � � |�
|� � d|j d�cddd� � S # 1 swxY w Y dS )z*Add or update a tracked query for the userNT)�ok�query_id)�
user_phone�
query_textr! )r r r �execr
r �wherer&