Update app.py
Browse files
app.py
CHANGED
|
@@ -10,6 +10,11 @@ model_name = "Dolphin-Phi"
|
|
| 10 |
# Load the chosen LLM model
|
| 11 |
llm = pipeline("text-generation", model=Dolphin-Phi)
|
| 12 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
# DSPy-based prompt generation
|
| 14 |
from dspy.agents import Agent
|
| 15 |
from dspy.utils import SentenceSplitter, SentimentAnalyzer, NamedEntityRecognizer
|
|
@@ -40,10 +45,10 @@ def dspy_generate_agent_prompts(prompt):
|
|
| 40 |
for sentence in sentences:
|
| 41 |
entities = ner.process(sentence)
|
| 42 |
for entity in entities:
|
| 43 |
-
if entity.label_ in ["FOOD", "ORG", "LOCATION"]: # Customize entity labels based on
|
| 44 |
extracted_entities.setdefault(entity.label_, []).append(entity.text)
|
| 45 |
|
| 46 |
-
# 4. Craft prompts for each agent
|
| 47 |
agent_prompts = []
|
| 48 |
|
| 49 |
# **Sentiment Analyzer Prompt:**
|
|
|
|
| 10 |
# Load the chosen LLM model
|
| 11 |
llm = pipeline("text-generation", model=Dolphin-Phi)
|
| 12 |
|
| 13 |
+
#Vectara config:
|
| 14 |
+
# customer_id =
|
| 15 |
+
# corpus_id =
|
| 16 |
+
# api_key =
|
| 17 |
+
|
| 18 |
# DSPy-based prompt generation
|
| 19 |
from dspy.agents import Agent
|
| 20 |
from dspy.utils import SentenceSplitter, SentimentAnalyzer, NamedEntityRecognizer
|
|
|
|
| 45 |
for sentence in sentences:
|
| 46 |
entities = ner.process(sentence)
|
| 47 |
for entity in entities:
|
| 48 |
+
if entity.label_ in ["FOOD", "ORG", "LOCATION"]: # Customize entity labels based on needs
|
| 49 |
extracted_entities.setdefault(entity.label_, []).append(entity.text)
|
| 50 |
|
| 51 |
+
# 4. Craft prompts for each agent (incomplete)
|
| 52 |
agent_prompts = []
|
| 53 |
|
| 54 |
# **Sentiment Analyzer Prompt:**
|