Skip to content

Commit

Permalink
isort main.py
Browse files Browse the repository at this point in the history
  • Loading branch information
abodhankar committed Aug 30, 2024
1 parent 6a64f6d commit 75118dd
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions examples/bots/llamaguard/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

from ingest import retriever
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.pydantic_v1 import BaseModel
Expand All @@ -21,8 +22,6 @@
from nemoguardrails import RailsConfig
from nemoguardrails.integrations.langchain.runnable_rails import RunnableRails

from ingest import retriever

llm = ChatNVIDIA(
model="meta/llama-3.1-70b-instruct",
temperature=0,
Expand Down Expand Up @@ -64,7 +63,7 @@ class Question(BaseModel):
chain_with_llamaguard = chain_with_llamaguard.with_types(input_type=Question)

print(
f"Hello, I am an AI Assistant that can answer any questions on NVIDIA AI Enterprise."
"Hello, I am an AI Assistant that can answer any questions on NVIDIA AI Enterprise."
)
question = input("How can I help you today? \n\n")
print("\n")
Expand Down

0 comments on commit 75118dd

Please sign in to comment.