Skip to content

Commit

Permalink
Code cleaning
Browse files Browse the repository at this point in the history
  • Loading branch information
nikolamilosevic86 committed Jun 6, 2024
1 parent 5e17fb1 commit 8be4884
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 6 deletions.
8 changes: 4 additions & 4 deletions api.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from fastapi import FastAPI
import qdrant_client
#import qdrant_client
from langchain_community.embeddings import HuggingFaceEmbeddings
from langchain_core.messages import HumanMessage
#from langchain_core.messages import HumanMessage
from langchain_qdrant import Qdrant
from qdrant_client import QdrantClient
from pydantic import BaseModel
Expand All @@ -10,7 +10,7 @@
import environment_var
import os
from openai import OpenAI
from langgraph.graph import END, MessageGraph
#from langgraph.graph import END, MessageGraph

class Item(BaseModel):
query: str
Expand All @@ -33,7 +33,7 @@ def __init__(self, query: str) -> None:
if environment_var.nvidia_key !="":
client_ai = OpenAI(
base_url="https://integrate.api.nvidia.com/v1",
api_key="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"
api_key=environment_var.nvidia_key
)
use_nvidia_api = True
elif use_quantized:
Expand Down
1 change: 0 additions & 1 deletion index.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
import qdrant_client
import PyPDF2
from os import listdir
from os.path import isfile, join,isdir
Expand Down
1 change: 0 additions & 1 deletion user_interface.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
import re
from pathlib import Path
import streamlit as st
import requests
import json
Expand Down

0 comments on commit 8be4884

Please sign in to comment.