sarim commited on
Commit
50d1ce2
·
1 Parent(s): 279a6a3

improve promopt

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import asyncio
 
2
  import streamlit as st
3
  from pydantic_ai import Agent
4
  from pydantic_ai.models.groq import GroqModel
@@ -9,7 +10,6 @@ import os
9
  api_key = os.getenv("API_KEY")
10
  data = []
11
 
12
- #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
13
 
14
  model = GroqModel('llama-3.1-70b-versatile', api_key = api_key)
15
 
@@ -31,11 +31,20 @@ async def ppt_content(data):
31
  listOfString = split_long_string("".join(data))
32
  print(len(listOfString))
33
 
 
 
 
 
 
34
 
35
  result_1 = agent.run_sync(user_prompt = listOfString[0])
36
  result_2 = agent.run_sync(user_prompt = listOfString[1],message_history=result_1.new_messages())
37
- result_3 = agent.run_sync(user_prompt = listOfString[1],message_history=result_2.new_messages())
38
- print(result_3.data)
 
 
 
 
39
 
40
  def ai_ppt(data):
41
  asyncio.run(ppt_content(data=data))
 
1
  import asyncio
2
+ from pydantic_ai.result import ResultData, RunResult
3
  import streamlit as st
4
  from pydantic_ai import Agent
5
  from pydantic_ai.models.groq import GroqModel
 
10
  api_key = os.getenv("API_KEY")
11
  data = []
12
 
 
13
 
14
  model = GroqModel('llama-3.1-70b-versatile', api_key = api_key)
15
 
 
31
  listOfString = split_long_string("".join(data))
32
  print(len(listOfString))
33
 
34
+ # for x in listOfString:
35
+ # result: RunResult[str] = RunResult(_all_messages:[])
36
+ # result = agent.run_sync(user_prompt = x,message_history = result.new_messages())
37
+
38
+
39
 
40
  result_1 = agent.run_sync(user_prompt = listOfString[0])
41
  result_2 = agent.run_sync(user_prompt = listOfString[1],message_history=result_1.new_messages())
42
+ result_3 = agent.run_sync(user_prompt = listOfString[2],message_history=result_2.new_messages())
43
+ result_4 = agent.run_sync(user_prompt = listOfString[3],message_history=result_3.new_messages())
44
+ result_5 = agent.run_sync(user_prompt = listOfString[4],message_history=result_4.new_messages())
45
+ result_6 = agent.run_sync(user_prompt = listOfString[5],message_history=result_5.new_messages())
46
+
47
+ print(result_6.data)
48
 
49
  def ai_ppt(data):
50
  asyncio.run(ppt_content(data=data))