diff --git a/interviewkit/interview.py b/interviewkit/interview.py index a5d579b..0593e87 100644 --- a/interviewkit/interview.py +++ b/interviewkit/interview.py @@ -1,6 +1,5 @@ from random import choice from interviewee import Interviewee -import openai from enum import StrEnum, auto @@ -40,20 +39,15 @@ def stop_transcription(self) -> None: self.transcript = " ".join(self.transcript) def suggest_questions(self) -> str: - """Use the OpenAI API to suggest questions based on the transcript content.""" + """suggest questions based on the transcript content.""" suggested_questions = self.generate_questions(self.transcript) return suggested_questions def generate_questions(self, content: str) -> list[str]: - """Call the OpenAI API to generate questions, and return them.""" - response = openai.Completion.create( - engine="text-davinci-002", - prompt=f"Generate interview questions based on the following transcript:\n{content}\n", - max_tokens=50, # Adjust the max tokens as needed - n=5, # Number of questions to generate - stop=None, - temperature=0.7, # Adjust the temperature for creativity - ) + """generate questions, and return them.""" + + #TODO + response = None # Extract and return the generated questions generated_questions = [choice["text"].strip() for choice in response.choices] diff --git a/interviewkit/main.py b/interviewkit/main.py index c960be0..05e0b43 100644 --- a/interviewkit/main.py +++ b/interviewkit/main.py @@ -1,17 +1,11 @@ import os -import openai - from interview import Interview from interviewee import Interviewee from transcript import Transcript from settings import Settings - -# Set the OpenAI API key settings = Settings() -openai.api_key = settings.OPENAI_API_KEY - if __name__ == "__main__": interviewee = Interviewee("John Doe", 60, "Male") diff --git a/interviewkit/settings.py b/interviewkit/settings.py index cd143e2..8a630f2 100644 --- a/interviewkit/settings.py +++ b/interviewkit/settings.py @@ -2,4 +2,5 @@ class Settings(BaseSettings): - OPENAI_API_KEY: str + #TODO + pass \ No newline at end of file