cj-dev-code/semantic_search
0
1o
2 v��h$ � @ s d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z d dl3m Z d dl m
Z
d dlZd dlZd d lmZ d d4lmZ d dlZe� e�� Ze�d �Ze�d �Zd
Zeeee e�dd�Z e� Z dd� Z e� Z G dd� de�Z e �!d�de fdd��Z"ej#e edd�Z dS )� )�FastAPI)� BaseModel)� load_dotenv)� SearchClient)�Vector)�AzureKeyCredential)�pipelineN)�OpenAI)�ifaceZAZURE_AAIS_ENDPTZAZURE_AAIS_KEY�mainz2023-07-01-Preview)�endpoint�5index_nameZ6credential� api_versionc C s t jj| ddd�}|jd jS )Nztext-embedding-3-largei )�input�modelZ7dimensionsr )�clienttZ8embeddings�create�data� embedding)�text�response� r �k/home/moonlight/Desktop/Daily_Tools/Github_Projects/CJ/semantic_searcher/semantic_search/backend/backend.py�embed% s � r c @ s e Zd ZU eed< dS )� QueryRequest�queryN)�__name__�9__module__� __qualname__�str�__annotations__r r r r r 3 s 10 r z/search�
query_requestc � sZ �| j }t|�}dd l}t|�tj�� tjd t|ddd�gddgd�}dd d11� |D �iS ) Nr � r )�value�k�fieldsr �source)Z search_textZvectors�select�resultsc S s g | ] }|d |d d��qS )r r& )r r& r )�.0�rr r r �12<listcomp>M s ��z!vector_search.<locals>.<listcomp>)r r �inspect�print� signature�client�searchr )r! r Zvectorr, r( r r r �
vector_search6 s& ������r1 � )�path)$Zfastapir Zpydanticr �dotenvr Zazure.search.documentsr Z(azure.search.documents._generated.modelsr Zazure.core.credentialsr Z transformersr Zvoyageai�osZopenair Z frontend.appr13 ZgradioZgr�ClientZvo�getenvr �keyr
r/ r r �appr Zpostr1 Zmount_gradio_appr r r r �<module> s< 1415�