Spaces:
Running
Running
improve promopt
Browse files
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[
|
| 38 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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))
|