Create module for llm api

meal_planner_api/connect_llm_api.py → llm_api/connect.py

@@ -1,6 +1,6 @@
-import utils
+from meal_planner_api import utils
 import google.generativeai as genai
-from config import API_KEY
+from meal_planner_api import config
 
 
 def generate_content(text: str):
@@ -10,6 +10,6 @@ def generate_content(text: str):
 
 
 def connect():
-    genai.configure(api_key=API_KEY)
+    genai.configure(api_key=config.API_KEY)
     model = genai.GenerativeModel('gemini-pro')
     return model

meal_planner_api/create_meal_plan.py

@@ -1,4 +1,4 @@
-import connect_llm_api
+from llm_api import connect
 from utils import to_markdown
 
 
@@ -7,7 +7,7 @@ def generate(parameters):
     prompt = create_prompt(**parameters)
 
     # Call your API or logic to generate the meal plan based on the prompt
-    meal_plan = connect_llm_api.generate_content(prompt)
+    meal_plan = connect.generate_content(prompt)
 
     return to_markdown(prompt+'/n*******/n'+meal_plan)