GroqCityVisitRecommender.predict:v0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import weave
@weave.op()
def predict(self, city: str) -> str:
system_message = {
"role": "system",
"content": "You are a helpful assistant meant to suggest places to visit in a city",
}
user_message = {"role": "user", "content": city}
chat_completion = self.groq_client.chat.completions.create(
messages=[system_message, user_message],
model=self.model,
)
return chat_completion.choices[0].message.content