Skip to content

Commit d2f961b

Browse files
committed
Added Qdrant Cloud configuration for deployment environment
1 parent 9515f23 commit d2f961b

File tree

2 files changed

+11
-16
lines changed

2 files changed

+11
-16
lines changed

ai.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -230,7 +230,17 @@ def review_application(job_description: str) -> ReviewedApplication:
230230

231231
def get_vector_store():
232232
embeddings = OpenAIEmbeddings(model="text-embedding-3-large", api_key=settings.OPENAI_API_KEY)
233-
vector_store = QdrantVectorStore.from_existing_collection(embedding=embeddings, collection_name="resumes", path="qdrant_store")
233+
if settings.PRODUCTION:
234+
vector_store = QdrantVectorStore.from_existing_collection(
235+
embedding=embeddings,
236+
collection_name="resumes",
237+
url=str(settings.QDRANT_URL),
238+
api_key=settings.QDRANT_API_KEY)
239+
else:
240+
vector_store = QdrantVectorStore.from_existing_collection(
241+
embedding=embeddings,
242+
collection_name="resumes",
243+
path="qdrant_store")
234244
return vector_store
235245

236246
def inmemory_vector_store():

main.py

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -111,21 +111,6 @@ async def api_close_job_post(job_post_id, db: Session = Depends(get_db)):
111111
db.commit()
112112
return jobPost
113113

114-
@app.get("/api/job-posts/{job_post_id}/recommend")
115-
async def api_recommend_resume(
116-
job_post_id,
117-
db: Session = Depends(get_db),
118-
vector_store = Depends(get_vector_store)):
119-
120-
job_post = db.get(JobPost, job_post_id)
121-
if not job_post:
122-
raise HTTPException(status_code=400)
123-
job_description = job_post.description
124-
recommended_resume = get_recommendation(job_description, vector_store)
125-
application_id = recommended_resume.metadata["_id"]
126-
job_application = db.get(JobApplication, application_id)
127-
return job_application
128-
129114
class JobPostForm(BaseModel):
130115
title : str
131116
description: str

0 commit comments

Comments
 (0)