adsjflkafj
This commit is contained in:
parent
521c85f726
commit
8d5aa4aad2
5 changed files with 13 additions and 8 deletions
|
|
@ -82,7 +82,7 @@ async def generate_ppt_outline(
|
|||
language: Optional[str] = None,
|
||||
content: Optional[str] = None,
|
||||
):
|
||||
model = get_nano_model()
|
||||
model = get_large_model()
|
||||
response_model = get_presentation_outline_model_with_n_slides(n_slides)
|
||||
|
||||
if not is_google_selected():
|
||||
|
|
|
|||
|
|
@ -1,6 +1,11 @@
|
|||
from models.presentation_layout import PresentationLayoutModel
|
||||
from models.presentation_outline_model import PresentationOutlineModel
|
||||
from utils.llm_provider import get_llm_client, get_nano_model, get_small_model
|
||||
from utils.llm_provider import (
|
||||
get_large_model,
|
||||
get_llm_client,
|
||||
get_nano_model,
|
||||
get_small_model,
|
||||
)
|
||||
from utils.get_dynamic_models import (
|
||||
get_presentation_structure_model_with_n_slides,
|
||||
)
|
||||
|
|
@ -62,7 +67,7 @@ async def generate_presentation_structure(
|
|||
) -> PresentationStructureModel:
|
||||
|
||||
client = get_llm_client()
|
||||
model = get_nano_model()
|
||||
model = get_large_model()
|
||||
response_model = get_presentation_structure_model_with_n_slides(
|
||||
len(presentation_outline.slides)
|
||||
)
|
||||
|
|
|
|||
|
|
@ -5,8 +5,8 @@ from models.presentation_layout import SlideLayoutModel
|
|||
from models.presentation_outline_model import SlideOutlineModel
|
||||
from utils.llm_provider import (
|
||||
get_google_llm_client,
|
||||
get_large_model,
|
||||
get_llm_client,
|
||||
get_nano_model,
|
||||
is_google_selected,
|
||||
)
|
||||
from utils.schema_utils import remove_fields_from_schema
|
||||
|
|
@ -55,7 +55,7 @@ def get_prompt_to_generate_slide_content(title: str, outline: str):
|
|||
async def get_slide_content_from_type_and_outline(
|
||||
slide_layout: SlideLayoutModel, outline: SlideOutlineModel
|
||||
):
|
||||
model = get_nano_model()
|
||||
model = get_large_model()
|
||||
|
||||
response_schema = remove_fields_from_schema(
|
||||
slide_layout.json_schema, ["__image_url__", "__icon_url__"]
|
||||
|
|
|
|||
|
|
@ -1,7 +1,7 @@
|
|||
from models.presentation_layout import PresentationLayoutModel, SlideLayoutModel
|
||||
from models.slide_layout_index import SlideLayoutIndex
|
||||
from models.sql.slide import SlideModel
|
||||
from utils.llm_provider import get_llm_client, get_nano_model, get_small_model
|
||||
from utils.llm_provider import get_large_model, get_llm_client
|
||||
|
||||
|
||||
def get_prompt_to_select_slide_layout(
|
||||
|
|
@ -42,7 +42,7 @@ async def get_slide_layout_from_prompt(
|
|||
) -> SlideLayoutModel:
|
||||
|
||||
client = get_llm_client()
|
||||
model = get_nano_model()
|
||||
model = get_large_model()
|
||||
|
||||
slide_layout_ids = list(map(lambda x: x.id, layout.slides))
|
||||
|
||||
|
|
|
|||
|
|
@ -92,7 +92,7 @@ def get_google_llm_client():
|
|||
def get_large_model():
|
||||
selected_llm = get_llm_provider()
|
||||
if selected_llm == LLMProvider.OPENAI:
|
||||
return "gpt-4.1"
|
||||
return "o3"
|
||||
elif selected_llm == LLMProvider.GOOGLE:
|
||||
return "gemini-2.0-flash"
|
||||
elif selected_llm == LLMProvider.OLLAMA:
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue