Address comments

run-llama · Feb 27, 2024 · 1dd4a53 · 1dd4a53
1 parent 2cd5a26
commit 1dd4a53
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/docs/module_guides/models/llms/modules.md b/docs/module_guides/models/llms/modules.md
@@ -84,6 +84,15 @@ maxdepth: 1
 /examples/llm/fireworks.ipynb
 ```
 
+## Friendli
+
+```{toctree}
+---
+maxdepth: 1
+---
+/examples/llm/friendli.ipynb
+```
+
 ## Gradient
 
 ```{toctree}

diff --git a/llama-index-integrations/llms/llama-index-llms-friendli/BUILD b/llama-index-integrations/llms/llama-index-llms-friendli/BUILD
@@ -1 +1,3 @@
-python_sources()
+poetry_requirements(
+    name="poetry",
+)
diff --git a/llama-index-integrations/llms/llama-index-llms-friendli/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-friendli/pyproject.toml
@@ -7,6 +7,13 @@ check-filenames = true
 check-hidden = true
 skip = "*.csv,*.html,*.json,*.jsonl,*.pdf,*.txt,*.ipynb"
 
+[tool.llamahub]
+contains_example = false
+import_path = "llama_index.llms.friendli"
+
+[tool.llamahub.class_authors]
+Friendli = "kooyunmo"
+
 [tool.mypy]
 disallow_untyped_defs = true
 exclude = ["_static", "build", "examples", "notebooks", "venv"]