afiqzudinhadi/linkedIn_auto_jobs_applier_with_AI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

check gemini

Browse source
This commit is contained in:
feder-cr 2024-09-13 23:44:34 +02:00
parent 9e21417e02
commit acfc990b87
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/llm/llm_manager.py
View file

@ -84,7 +84,7 @@ class GeminiModel(AIModel):
HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE
},max_output_tokens=3000)
})
def invoke(self, prompt: str) -> BaseMessage:
response = self.model.invoke(prompt)